Class LinearModule

include/engine/framework/modules/linear_module.h:22–38

struct LinearWeights {
    core::TensorValue weight;
    std::optional<core::TensorValue> bias;
};

class LinearModule {
public:
    explicit LinearModule(LinearConfig config);

    const LinearConfig & config() const noexcept;
    const core::ModuleSchema & schema() const noexcept;

    core::TensorValue build(
        core::ModuleBuildContext & ctx,
        const core::TensorValue & input,
        const LinearWeights & weights) const;

    static const core::ModuleSchema & static_schema() noexcept;

private:
    LinearConfig config_;
};