开发者
资源
[object Object][object Object]

torch_npu\csrc\core\npu\NPUFormat.h

[object Object]
[object Object]
[object Object]

获取指定格式的NPU空tensor,返回值类型Tensor,表示获取的空tensor。

[object Object]

sizes:IntArrayRef类型,获取tensor的size。

options:TensorOptions类型,获取tensor的可选信息,如dtype、device等。

acl_format:int64_t类型,指定获取的格式。

keep_format:bool类型,是否指定格式,true表示指定获取tensor的格式,false表示允许根据算子实际需求调整获取tensor的格式。

[object Object]
  • [object Object]Atlas 训练系列产品[object Object]
  • [object Object]Atlas A2 训练系列产品[object Object]
  • [object Object]Atlas A3 训练系列产品[object Object]
  • [object Object]Atlas 推理系列产品[object Object]