ToContiguousStrides

函数功能

根据已有的shape信息，按照连续规则去推导对应的stride信息。例如shape为[10, 20, 30]，那么推导出的stride为[600, 30, 1]。

函数原型

void ToContiguousStrides(const Shape &shape, FVector<int64_t, 25> &strides)

参数说明

返回值说明

无

约束说明

无

调用示例

// 生成一个shape信息为[1, 2, 3, 4, 5]的Shape对象，并生成它的stride数据。
void Func() {
    gert::Shape newShape;
    for (int64_t i = 1; i <= 5; i++) {
        newShape.AppendDim(i);
    }
    FVector<int64_t, 25> strides;
    ToContiguousStrides(newShape, strides);
}