mindspore.ops.Shape
- class mindspore.ops.Shape[源代码]
返回输入Tensor的shape。用于静态shape。
获取更多详情请查看
mindspore.ops.shape()
。- 支持平台:
Ascend
GPU
CPU
样例:
>>> input_x = Tensor(np.ones(shape=[3, 2, 1]), mindspore.float32) >>> shape = ops.Shape() >>> output = shape(input_x) >>> print(output) (3, 2, 1)