函数: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,错误码。
|
约束说明 |
无 |
父主题: aclmdlTensorDesc