shape

• paddle.fluid.layers.shape(input)[源代码]

shape层。

获得输入Tensor的shape。

• 参数：
• • input （Variable）- 输入的多维Tensor，数据类型为float32，float64，int32，int64。

返回： 一个Tensor，表示输入Tensor的shape。

返回类型： Variable(Tensor)。

代码示例：

import paddle.fluid as fluid
import numpy as np
 
inputs = fluid.layers.data(name="x", shape=[3, 100, 100], dtype="float32")
output = fluid.layers.shape(inputs)
 
exe = fluid.Executor(fluid.CPUPlace())
exe.run(fluid.default_startup_program())
 
img = np.ones((3, 100, 100)).astype(np.float32)
 
res = exe.run(fluid.default_main_program(), feed={'x':img}, fetch_list=[output])
print(res) # [array([  3, 100, 100], dtype=int32)]