（beta）torch_npu.npu_format_cast_

接口原型

torch_npu.npu_format_cast_(self, src) -> Tensor

功能描述

原地修改self张量格式，与src格式保持一致。src，即source tensor，源张量。

参数说明

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

支持的型号

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

调用示例

1
2
3
4
5

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