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

add transpose, start on transformer

Browse Source
This commit is contained in:
George Hotz
2020-12-27 16:59:12 -05:00
parent d379502c04
commit 2f1b2c0a3b
13 changed files with 91 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_tensor.py
View File

@@ -1,10 +1,8 @@
import numpy as np
import torch
import unittest
from tinygrad.tensor import Tensor, GPU, Device
from tinygrad.tensor import Tensor, GPU, ANE, Device
from extra.gradcheck import numerical_jacobian, jacobian, gradcheck
from .config import ANE
x_init = np.random.randn(1,3).astype(np.float32)
U_init = np.random.randn(3,3).astype(np.float32)