Method from_dims

src/framework/core/module.cpp:37–51

}  // namespace

TensorShape TensorShape::from_dims(std::initializer_list<int64_t> dims_init) {
    TensorShape shape;
    shape.rank = dims_init.size();
    if (shape.rank == 0 || shape.rank > kMaxTensorRank) {
        throw std::runtime_error("Tensor rank must be between 1 and 4");
    }

    size_t index = 0;
    for (const int64_t dim : dims_init) {
        shape.dims[index++] = dim;
    }

    ensure_positive_dims(shape);
    return shape;
}

int64_t TensorShape::at(size_t index) const {
    if (index >= rank) {