github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity
Files
02636ff62d95186dd541a245ef0df9cf418dedd2
tinygrad/test/test_tensor.py
Obada Khalili ee25f73283 Fix Tensor.mean to compute the mean correctly when 0-length axes are selected (#3318) 
* fix Tensor.mean to compute the mean correctly with 0-length axes are selected

* add a regression test

* rename sum variable to sum_t to avoid conflict with built it function

* refactor Tensor.mean to has less lines
2024-02-05 01:40:37 -05:00

18 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink