如何打印张量形状的值?

时间:2018-01-17 13:11:20

标签: python arrays multidimensional-array tensorflow tensor

我使用了函数

print tf.shape(image)

输出喜欢

Tensor("Shape:0", shape=(3,), dtype=int32, device=/device:CPU:0)

我想知道形状内部的值是什么(比如尺寸)。如何访问它进行打印?

1 个答案:

答案 0 :(得分:2)

tf.shape返回包含参数形状的张量。当其中一个维度动态时,这非常有用,即静态None

您可以使用image.shape(或image.get_shape())获取静态形状,也可以在会话中评估tf.shape(image)

另见this answer