函数：set_tensor_storage_format
产品支持情况
|
产品 |
是否支持 |
|---|---|
|
Atlas 350 加速卡 |
√ |
|
|
√ |
|
|
√ |
|
|
x |
|
|
√ |
|
|
√ |
功能说明
调用acl.create_tensor_desc接口创建Tensor描述信息后，可通过acl.set_tensor_storage_format设置Tensor的实际Format信息，当前主要用于PyTorch场景下。
此接口后续版本会废弃，请使用acl.set_tensor_format接口。
函数原型
- C函数原型
1aclError aclSetTensorStorageFormat(aclTensorDesc *desc, aclFormat format)
- python函数
1ret = acl.set_tensor_storage_format(desc, format)
参数说明
|
参数名 |
说明 |
|---|---|
|
desc |
int，aclTensorDesc类型的指针地址。 需提前调用acl.create_tensor_desc接口创建aclTensorDesc类型。 |
|
format |
int， 指定Tensor描述的格式。 |
父主题： aclmdlTensorDesc