at_npu::native::get_npu_storage_sizes

定义文件

torch_npu\csrc\core\npu\NPUFormat.h

API接口

std::vector<int64_t> at_npu::native::get_npu_storage_sizes(const at::Tensor& self)

功能描述

获取npu tensor的内存大小，返回值类型vector<int64_t>，表示获取的npu tensor内存大小。

参数说明

self：Tensor类型，待获取内存大小的tensor。