shape

shape层。

获得输入Tensor的shape。

  • 参数:
    • input (Variable)- 输入的多维Tensor,数据类型为float32,float64,int32,int64。

返回: 一个Tensor,表示输入Tensor的shape。

返回类型: Variable(Tensor)。

代码示例:

  1. import paddle.fluid as fluid
  2. import numpy as np
  3.  
  4. inputs = fluid.layers.data(name="x", shape=[3, 100, 100], dtype="float32")
  5. output = fluid.layers.shape(inputs)
  6.  
  7. exe = fluid.Executor(fluid.CPUPlace())
  8. exe.run(fluid.default_startup_program())
  9.  
  10. img = np.ones((3, 100, 100)).astype(np.float32)
  11.  
  12. res = exe.run(fluid.default_main_program(), feed={'x':img}, fetch_list=[output])
  13. print(res) # [array([ 3, 100, 100], dtype=int32)]