torch_npu.npu接口列表

本章节包含各种子模块接口，如随机数、内存管理等。npu模块支持torch.npu和torch_npu.npu两种调用方式，功能一致。

表1 torch_npu.npu API

[object Object][object Object]

[object Object]

表2 amp API

[object Object][object Object]

[object Object]

表3 Random Number Generator API

[object Object][object Object]

[object Object]

表4 NPU device API

[object Object][object Object]

[object Object]

表5 Memory management API

[object Object][object Object]

[object Object]

表6 aoe API

[object Object][object Object]

[object Object]

表7 Profiler API

[object Object][object Object]

[object Object]

表8 torch_npu Storage API

[object Object][object Object]

[object Object]

表9 NPU Tensor API(beta)

[object Object][object Object]

[object Object]