mirror of
https://github.com/MLSysBook/TinyTorch.git
synced 2026-05-10 16:38:39 -05:00
Introduces a Tensor class that wraps numpy arrays, enabling fundamental ML operations like addition, subtraction, multiplication, and division. Adds utility methods such as reshape, transpose, sum, mean, max, min, item, and numpy to the Tensor class. Updates tests to accommodate both scalar and Tensor results when checking mean values.