（beta）torch_npu.npu_format_cast

接口原型

torch_npu.npu_format_cast(self, acl_format) -> Tensor

功能描述

修改NPU张量的格式。

参数说明

• self (Tensor) - 输入张量。
• acl_format (Int) - 目标格式。数据格式具体可参考《CANN AscendCL应用软件开发指南 (Python)》中“aclFormat”章节。此处需输入数字，例如2，即代表ACL_FORMAT_ND格式。

支持的型号

• Atlas 训练系列产品
• Atlas A2 训练系列产品
• Atlas A3 训练系列产品
• Atlas 推理系列产品

调用示例

1
2
3
4
5
6

>>> x = torch.rand(2, 3, 4, 5).npu()
>>> torch_npu.get_npu_format(x)
0
>>> x1 = torch_npu.npu_format_cast(x, 2)
>>> torch_npu.get_npu_format(x1)
2