如何切分一个无维张量？

Question

我只是用TensorFlow来实现一个CNN模型DVF:https://github.com/liuziwei7/voxel-flow。在

模型的输出是'deconv4'，形状为[batch_size，256，256,3]，然后我需要使用命令'flow=tf.切片（deconv4.outputs，[0,0,0,0]，[batch_size，256，256，2]）。在

但是，如果“批处理大小”为“无”，如何切片“流”张量？在

提前谢谢。在

“deconv4”的形状是[？，256256,3]我想得到形状为[？，256256,2]来自“deconv4”。在

deconv4 = Conv2d(deconv3_bn_relu, 3, [5, 5], act=tf.tanh, padding='SAME', W_init=w_init, name='deconv4')

    #################### Calculate Voxel Flow based on the 'deconv4' ############################
    flow = tf.slice(deconv4.outputs, [0,0,0,0], [batch_size, 256, 256, 2])

“流”的形状应该是[？，256256,2]。但我不知道如何获得它。在

Answer 1

您应该能够用None替换批处理的大小，这样就可以了。在

或者，这个：tf.shape(x)[0]将给您一个带有批量大小的变量张量。在