ifkash/smoltorch
1
0
Fork 0
mirror of https://github.com/kashifulhaque/smoltorch.git synced 2025-12-06 07:02:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Pre-release

Browse Source
This commit is contained in:
Kashif
2025-11-17 21:49:37 +05:30
parent a6e32fcc71
commit 9a1bc55c27
4 changed files with 132 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
nanotorch/tensor.py
View File

@@ -147,6 +147,15 @@ class Tensor:
out._backward = _backward
return out
def log(self) -> 'Tensor':
out = Tensor(np.log(self.data), (self, ), 'log')
def _backward():
self.grad += (1 / self.data) * out.grad
out._backward = _backward
return out
def backward(self):
# build topological order
topo = []
@@ -187,3 +196,14 @@ class Tensor:
out._backward = _backward
return out
def sigmoid(self) -> 'Tensor':
sig = 1 / (1 + np.exp(-self.data))
out = Tensor(sig, (self, ), 'sigmoid')
def _backward():
self.grad += sig * (1 - sig) * out.grad
out._backward = _backward
return out