github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-19 02:44:40 -05:00
Code Issues Packages Projects Releases Wiki Activity

Fixed test suite for Clip (#912)

Browse Source 
* Fixed test suite for Clip

* fixed issue with clip when taking large negative numbers as min

* Remove typings
This commit is contained in:
Steven Anderson
2023-06-04 09:01:01 -07:00
committed by GitHub
parent 0b88c5f923
commit 657e642e3a
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/tensor.py
View File

@@ -478,7 +478,7 @@ class Tensor:
def sqrt(self): return self.pow(0.5)
def rsqrt(self): return self.pow(-0.5)
def square(self): return self*self
def clip(self, min_, max_): return ((self-min_).relu()+min_) - (self-max_).relu()
def clip(self, min_, max_): return self.maximum(min_).minimum(max_)
def abs(self): return self.relu() + (-self).relu()
def sign(self): return self / (self.abs() + 1e-10)
def reciprocal(self): return 1.0/self