合作咨询
我们的专家服务团队将竭诚为您提供专业的合作咨询服务
获取当前Tensor运行时的shape大小,即此Tensor中包含的元素的数量。
int64_t GetShapeSize() const
无。
返回执行时shape的大小。
无
Tensor tensor{{{8, 3, 224, 224}, {16, 3, 224, 224}}, // shape
{ge::FORMAT_ND, ge::FORMAT_FRACTAL_NZ, {}}, // format
kOnDeviceHbm, // placement
ge::DT_FLOAT16, // dt
nullptr};
auto shape_size = tensor.GetShapeSize(); // 16*3*224*224 合作咨询
我们的专家服务团队将竭诚为您提供专业的合作咨询服务
解决方案
精准高效的一站式服务支持,助力开发者商业成功