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

Tensor.logsigmoid (#10955)

Browse Source
This commit is contained in:
chenyu
2025-06-24 11:16:14 -04:00
committed by GitHub
parent f15247d2d2
commit 18e264a449
4 changed files with 19 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
test/test_ops.py
View File

@@ -946,6 +946,11 @@ class TestOps(unittest.TestCase):
self.assertAlmostEqual(sigmoid(x)[0].gradient(x)[0].item(), 0.0)
x = Tensor([-300.0])
self.assertAlmostEqual(sigmoid(x)[0].gradient(x)[0].item(), 0.0)
def test_logsigmoid(self):
helper_test_op([(45,65)], torch.nn.functional.logsigmoid, Tensor.logsigmoid)
helper_test_op([()], torch.nn.functional.logsigmoid, Tensor.logsigmoid)
def test_hardsigmoid(self):
helper_test_op([(45,65)], torch.nn.functional.hardsigmoid, Tensor.hardsigmoid)
helper_test_op([()], torch.nn.functional.hardsigmoid, Tensor.hardsigmoid)