at_npu::native::npu_format_cast

定义文件

torch_npu\csrc\core\npu\NPUFormat.h

API接口

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类型，待转换的格式。