(beta)at_npu::native::npu_format_cast
定义文件
torch_npu\csrc\core\npu\NPUFormat.h
接口原型
at::Tensor at_npu::native::npu_format_cast(const at::Tensor& self, int64_t acl_format)
功能描述
npu tensor格式转换,返回值类型Tensor,表示转换后的tensor。
参数说明
self:Tensor类型,待转换格式的tensor。
acl_format:int64_t类型,待转换的格式。
父主题: C++ API