合作咨询
我们的专家服务团队将竭诚为您提供专业的合作咨询服务
根据算子输出索引获取对应的输出shape指针。这里的输出索引是指算子实例化后实际的索引,不是原型定义中的索引。
const StorageShape *GetOutputShape(size_t index) const;
| 参数 | 输入/输出 | 说明 |
|---|---|---|
| index | 输入 | 算子输出索引,从0开始计数。 |
无
ge::graphStatus Tiling4ConcatD(TilingContext* context) {
auto out_shape = context->GetOutputShape(0);
// ...
} 合作咨询
我们的专家服务团队将竭诚为您提供专业的合作咨询服务
解决方案
精准高效的一站式服务支持,助力开发者商业成功