aparler/PyNorch
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix transpose cuda

Browse source
This commit is contained in:
lucasdelimanogueira 2024-05-16 19:11:15 -03:00
parent c11cfd7f22
commit 96a9ed57ff
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
norch/tensor.py
View file

@ -698,7 +698,7 @@ class Tensor:
def transpose(self, axis1, axis2, requires_grad=None):
def transpose(self, axis1, axis2):
if axis1 < 0:
axis1 = self.ndim + axis1
if axis2 < 0: