昇腾社区首页
中文
注册
开发者
下载

GetPlacement

函数功能

获取Tensor的placement。

函数原型

1
TensorPlacement GetPlacement() const

参数说明

返回值说明

返回tensor的placement。

关于TensorPlacement类型的定义,请参见TensorPlacement

约束说明

调用示例

1
2
3
4
5
6
TensorV2 tensor{{{8, 3, 224, 224}, {16, 3, 224, 224}},       // shape              
                {ge::FORMAT_ND, ge::FORMAT_FRACTAL_NZ, {}},  // format              
                kFollowing,                                  // placement              
                ge::DT_FLOAT16,                              //dt              
                nullptr};
auto placement = tensor.GetPlacement(); // kFollowing