void SetReshapeDimensions(Dims dimensions) noexcept;

函数功能

设置reshape操作的目标维度。

函数原型

void SetReshapeDimensions(Dims dimensions) noexcept;

约束说明

• 该接口只支持调用一次，ShuffleLayer可以只调用这一个操作来实现Reshape；也可以组合调用其他API，注意在有多个操作时，只允许在SetFirstTranspose之后和SetSecondTranspose之前调用。
• 维度不能为空，最多只能包含一个负数，即-1，该维度会在内部进行推导，满足reshape前后的Tensor元素总数相同。
• 维度支持传0，则该轴的实际目标维度会直接拷贝输入张量对应轴的维度。

参数说明

返回值说明

无