github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-23 05:48:08 -05:00
Code Issues Packages Projects Releases Wiki Activity

that's simpler

Browse Source
This commit is contained in:
George Hotz
2022-09-18 20:40:46 -04:00
parent 57e804a9bf
commit a8aa1f9589
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/tensor.py
View File

@@ -210,7 +210,7 @@ class Tensor:
def sum(self, axis=None, keepdim=False): return self._reduce(Tensor._sum, axis, keepdim)
def max(self, axis=None, keepdim=False): return self._reduce(Tensor._max, axis, keepdim)
def min(self, axis=None, keepdim=False): return -((-self)._reduce(Tensor._max, axis, keepdim))
def min(self, axis=None, keepdim=False): return -((-self).max(axis=axis, keepdim=keepdim))
def mean(self, axis=None, keepdim=False):
out = self.sum(axis=axis, keepdim=keepdim)