geohot/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Actions Activity

Simplify more (#117)

Browse Source 
Co-authored-by: holonomicjl <[email protected]>
This commit is contained in:
adamritter
2020-11-14 06:15:31 -08:00
committed by GitHub
co-authored by holonomicjl
parent 28474949b8
commit 55d93017e4
1 changed files with 4 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/opsgpu.py
+4 -8
View File
@@ -314,14 +314,10 @@ class Reshape(Function):
@staticmethod
def forward(ctx, x, shape):
ctx.save_for_backward(x.shape)
# I'm sorry for this code
tsum = functools.reduce(lambda x,y: x*y, (s for s in shape if s != -1), 1)
shape = tuple(np.prod(x.shape) // tsum if s == -1 else s for s in shape)
assert np.prod(x.shape) == np.prod(shape)
x = unary_op(ctx, 'a', x)
x.shape = shape
return x
r = unary_op(ctx, 'a', x)
r.shape = tuple(-np.prod(x.shape) // np.prod(shape) if s == -1 else s for s in shape)
assert np.prod(x.shape) == np.prod(r.shape)
return r
@staticmethod
def backward(ctx, grad_output):