BroadcastInferShape

函数功能

对于满足broadcast关系的两个shape，推导他们broadcast后的shape。例如[1, 10]与[2, 1] broadcast后的shape为[2, 10]。

函数原型

bool BroadcastInferShape(const Shape &self, const Shape &other, shape &broadcastShape)

参数说明

返回值说明

当self与other满足broadcast关系时，返回true，否则返回false。

约束说明

无

调用示例

// 生成shape为[2， 1]和[2, 10]的两个Shape对象，获取broadcast后的shape。
void Func() {
    gert::Shape shapeA;
    shapeA.AppendDim(1);
    shapeA.AppendDim(2);
    gert::Shape shapeB;
    shapeB.AppendDim(10);
    shapeB.AppendDim(2);
    gert::Shape shapeBrc;
    bool isBrc = BroadcastInferShape(shapeA, shapeB, shapeBrc);
}