we have matmul too

This commit is contained in:
George Hotz
2020-10-27 08:57:17 -07:00
parent f9788eba14
commit 8d75c9e4c4

View File

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