set_format

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

Sets the tensor format.

Prototype

1
set_format(format: Format) -> Tensor

Parameters

Parameter

Input/Output

Description

format

Input

Format to be set. For details, see Format.

Returns

(Tensor) Returns itself to support chain calls.

Restrictions

  • If format is not of the Format type, TypeError is thrown.
  • If the setting fails, RuntimeError is thrown.