如何在TensorFlow中清除tf.Tensor的形状信息？

Question

假设在训练网络期间，我们将所有图像的大小调整为512 * 512，因此可能有一个名为input:0的tf.Tensor input:0 ，形状为(batch_size, 512, 512, 3) tf.Tensor (batch_size, 512, 512, 3) 。

但是，在进行预测时，可以将多种尺寸的图像输入网络。 因此张量input:0的形状input:0应该类似于(batch_size, None, None, 3) ，因为图像的大小是未知的。

因此，如果我有一个Tensor of shape (batch_size, 512, 512, 3) ，我该如何“重塑”它(batch_size, None, None, 3) ？ 我试过了

inputs=tf.reshape(inputs, (batch_size, tf.shape(inputs)[1], tf.shape(inputs)[2], 3)

但输出仍然是形状(batch_size, 512, 512, 3) 。

Answer 1

我不相信你可以调整/重新调整神经网络中的权重/偏差项。 但是将图像输入的大小调整为512 * 512非常容易。 你考虑过吗？