set_shape
Applicability
Product |
Supported |
|---|---|
Atlas 350 Accelerator Card |
√ |
√ |
|
√ |
|
√ |
|
√ |
|
√ |
Function Usage
Sets the tensor shape.
Prototype
1 | set_shape(shape: List[int]) -> TensorHolder |
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
shape |
Input |
Shape dimension list. |
Returns
(TensorHolder) Returns itself to support chain calls.
Restrictions
- If the shape is not an integer list, TypeError is thrown.
- If all shape dimensions are not integers, TypeError is thrown.
- If the operation fails, RuntimeError is thrown.
Parent topic: TensorHolder