如何获取Tensorflow张量尺寸（形状）作为int值？

如何获取Tensorflow张量尺寸（形状）作为int值？

要将形状显示为整数列表，请执行

tensor.get_shape().as_list()

。

要完成

tf.shape()

通话，请尝试

tensor2 = tf.reshape(tensor,tf.TensorShape([num_rows*num_cols, 1]))

。或者，您可以直接

tensor2 = tf.reshape(tensor,tf.TensorShape([-1, 1]))

在可以推断出其第一维的地方进行 *** 作。