文档
注册

函数:set_tensor_storage_shape

此接口后续版本会废弃,请使用acl.set_tensor_shape接口。

C函数原型

aclError aclSetTensorStorageShape(aclTensorDesc *desc, int numDims, const int64_t *dims)

Python函数

ret = acl.set_tensor_storage_shape(desc, dims)

函数功能

调用acl.create_tensor_desc接口创建Tensor描述信息后,可通过acl.set_tensor_storage_shape设置Tensor的实际“dims”信息,当前主要用于PyTorch场景下。

Atlas 训练系列产品,不支持该接口。

输入说明

desc:int,aclTensorDesc类型的指针地址。

需提前调用acl.create_tensor_desc接口创建aclTensorDesc类型。

dims:list,要设置的“dims”维度数。

返回值说明

ret:int,错误码。

  • 返回0表示成功。
  • 返回其它值表示失败。

约束说明

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

当前产品无相关内容

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