Exemple #1
0
import tinygrad
from tinygrad.tensor import Tensor

x = Tensor.eye(3)
y = Tensor([[2.0, 0, -2.0]])
z = y.matmul(x).sum()
z.backward()

# print(x.grad)  # dz/dx
# print(y.grad)  # dz/dy
Exemple #2
0
import os

os.environ['GPAPI'] = 'cuda'

from tinygrad.tensor import Tensor

x = Tensor.eye(3).gpu()
y = Tensor([[2.0, 0, -2.0]]).gpu()
z = y.matmul(x)
print(z.cpu())
z = z.sum()
print(z.cpu())
z.backward()
print(x.grad.cpu())  # dz/dx
print(y.grad.cpu())  # dz/dy
print()
x1 = Tensor.eye(3)
y1 = Tensor([[2.0, 0, -2.0]])
z1 = y1.matmul(x1)
print(z1)
z1 = z1.sum()
print(z1)
z1.backward()
print(x1.grad)  # dz/dx
print(y1.grad)  # dz/dy