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

update Tensor.maximum (#8992)

Browse Source 
now it's just broadcast and UOp.maximum
This commit is contained in:
chenyu
2025-02-09 21:26:27 -05:00
committed by GitHub
parent 88add71c25
commit 9119716761
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tinygrad/tensor.py
View File

@@ -3307,9 +3307,7 @@ class Tensor(SimpleMathTrait):
print(Tensor([-1, 2, 3]).maximum(Tensor([-4, -2, 9])).numpy())
```
"""
# NOTE: the mid-point is for backward, revisit after new gradient API
if self.is_floating_point(): return (self<x).detach().where(x, (self==x).detach().where(((self * 0.5 + x * 0.5).cast(self.dtype)), self))
return (self<x).detach().where(x, self)
return self._apply_broadcasted_uop(UOp.maximum, x)
def minimum(self, x:Union[Tensor, ConstType]) -> Tensor:
"""