代码之家  ›  专栏  ›  技术社区  ›  Jame

如何打印张量的形状值?

  •  1
  • Jame  · 技术社区  · 6 年前

    我用过这个函数

    print tf.shape(image)
    

    输出如下

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

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

    1 回复  |  直到 6 年前
        1
  •  2
  •   Maxim    6 年前

    tf.shape 返回包含参数形状的张量。当其中一个维度为 动态 ,即is None 静态。

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

    另请参见 this answer .