(beta)at_npu::native::empty_with_format
定义文件
torch_npu\csrc\core\npu\NPUFormat.h
接口原型
at::Tensor at_npu::native::empty_with_format(c10::IntArrayRef sizes, const c10::TensorOptions& options, int64_t acl_format, bool keep_format = false)
功能描述
获取指定格式的npu 空tensor,返回值类型Tensor,表示获取的空tensor。
参数说明
sizes:IntArrayRef类型,获取tensor的size。
options:TensorOptions类型,获取tensor的可选信息,如dtype、device等。
acl_format:int64_t类型,指定获取的格式。
keep_format:bool类型,是否指定格式,true表示指定获取tensor的格式,false表示允许根据算子实际需求调整获取tensor的格式。
父主题: C++ API