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)