文档
注册

beta)at_npu::native::get_npu_format

定义文件

torch_npu\csrc\core\npu\NPUFormat.h

接口原型

int64_t at_npu::native::get_npu_format(const at::Tensor& self)

功能描述

获取npu tensor格式信息,返回值类型int64_t,表示获取的npu tensor格式信息。

该接口通常配合申请npu私有格式内存empty_with_format使用。

参数说明

self:Tensor类型,待获取格式信息的tensor。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词