colin
/
PyorchHack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

更新 hack_tensor.md

This commit is contained in:
colin 2024-03-06 19:39:56 +08:00
parent 36a5d796b6
commit 4579e1351c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
hack_tensor.md
View File

@ -4,7 +4,7 @@ python3.10/site-packages/torch/_tensor.py
```
def __repr__(self, *, tensor_contents=None):
return "Tensor %s" % (str(self.size()))
return "Tensor " + str(self.size()) + " " + self.device.type + " " + str(self.dtype)
if has_torch_function_unary(self):
return handle_torch_function(
Tensor.__repr__, (self,), self, tensor_contents=tensor_contents