github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-22 21:38:10 -05:00
Code Issues Packages Projects Releases Wiki Activity

remove extra copy on output

Browse Source
This commit is contained in:
George Hotz
2023-03-01 21:19:44 -08:00
parent b442e75c7a
commit 201d9a2d58
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/tensor.py
View File

@@ -87,7 +87,7 @@ class Tensor:
return self
def detach(self): return Tensor(self.lazydata, device=self.device, requires_grad=False)
def numpy(self) -> np.ndarray: return np.array(self.lazydata.toCPU())
def numpy(self) -> np.ndarray: return self.lazydata.toCPU()
# TODO: if things are realized this won't work
def to_(self, device:str):