github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-07 03:00:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

log and exp are first class ops

Browse Source
This commit is contained in:
George Hotz
2020-12-28 10:00:30 -05:00
parent ffff98db78
commit 593233b668
6 changed files with 56 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_ops.py
View File

@@ -67,6 +67,10 @@ class TestOps(unittest.TestCase):
helper_test_op([(45,65)], lambda x: torch.nn.functional.leaky_relu(x,0.01), Tensor.leakyrelu, device=self.device)
def test_abs(self):
helper_test_op([(45,65)], lambda x: torch.abs(x), Tensor.abs, device=self.device)
def test_log(self):
helper_test_op([(45,65)], lambda x: torch.log(x), Tensor.log, device=self.device)
def test_exp(self):
helper_test_op([(45,65)], lambda x: torch.exp(x), Tensor.exp, device=self.device)
def test_sigmoid(self):
helper_test_op([(45,65)], lambda x: x.sigmoid(), Tensor.sigmoid, device=self.device)
def test_dot(self):