Class ReshapeModule

include/engine/framework/modules/structural_modules.h:14–25

};

class ReshapeModule {
public:
    explicit ReshapeModule(ReshapeConfig config);

    const ReshapeConfig & config() const noexcept;
    const core::ModuleSchema & schema() const noexcept;
    core::TensorValue build(core::ModuleBuildContext & ctx, const core::TensorValue & input) const;
    static const core::ModuleSchema & static_schema() noexcept;

private:
    ReshapeConfig config_;
};

struct TransposeConfig {
    std::array<int, core::kMaxTensorRank> axes = {0, 1, 2, 3};