commaai
/
tinygrad
mirror of https://github.com/commaai/tinygrad.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

remove extra copy on output

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
Show Stats 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):