get_shape

Applicability

Product

Supported

Atlas 350 Accelerator Card

Atlas A3 training product/Atlas A3 inference product

Atlas A2 training product/Atlas A2 inference product

Atlas 200I/500 A2 inference product

Atlas inference product

Atlas training product

Function Usage

Obtains the tensor shape.

Prototype

1
get_shape() -> List[int]

Parameters

None

Returns

(List[int]) Shape list.

Restrictions

If the obtaining fails, RuntimeError is thrown.