torch_npu\csrc\core\npu\NPUFormat.h
[object Object]
获取指定格式的NPU空tensor,返回值类型Tensor,表示获取的空tensor。
sizes:IntArrayRef类型,获取tensor的size。
options:TensorOptions类型,获取tensor的可选信息,如dtype、device等。
acl_format:int64_t类型,指定获取的格式。
keep_format:bool类型,是否指定格式,true表示指定获取tensor的格式,false表示允许根据算子实际需求调整获取tensor的格式。
- [object Object]Atlas 训练系列产品[object Object]
- [object Object]Atlas A2 训练系列产品[object Object]
- [object Object]Atlas A3 训练系列产品[object Object]
- [object Object]Atlas 推理系列产品[object Object]