Function flatten_to_matrix_shape

src/framework/modules/linear_module.cpp:29–35

};

core::TensorShape flatten_to_matrix_shape(const core::TensorShape & shape) {
    if (shape.rank == 1) {
        return core::TensorShape::from_dims({1, shape.last_dim()});
    }

    return core::TensorShape::from_dims({shape.prefix_elements(), shape.last_dim()});
}

void validate_weight_shape(const LinearConfig & config, const LinearWeights & weights) {
    core::validate_shape(