mindspore.ops.shape

mindspore.ops.shape(input_x)[source]

Returns the shape of the input tensor. This operation is used in static shape cases.

static shape: A shape that can be obtained without running the graph. It is an inherent property of tensor and may be unknown. The static shape information can be completed by artificial setting. No matter what the input of the graph is, the static shape is not affected.

Parameters

input_x (Tensor) – The shape of tensor is \((x_1, x_2, ..., x_R)\).

Returns

tuple[int], the output tuple is constructed by multiple integers, \((x_1, x_2, ..., x_R)\).

Raises

TypeError – If input_x is not a Tensor.

Supported Platforms:

Ascend GPU CPU

Examples

>>> input_x = Tensor(np.ones(shape=[3, 2, 1]), mindspore.float32)
>>> output = ops.shape(input_x)
>>> print(output)
(3, 2, 1)