we have matmul too

This commit is contained in:
George Hotz 2020-10-27 08:57:17 -07:00
parent f9788eba14
commit 8d75c9e4c4
1 changed files with 1 additions and 1 deletions

View File

@ -25,7 +25,7 @@ from tinygrad.tensor import Tensor
x = Tensor.eye(3)
y = Tensor([[2.0,0,-2.0]])
z = y.dot(x).sum()
z = y.matmul(x).sum()
z.backward()
print(x.grad) # dz/dx