From 96a9ed57ffb9ef392bb0a05c06b520997dd48edd Mon Sep 17 00:00:00 2001 From: lucasdelimanogueira Date: Thu, 16 May 2024 19:11:15 -0300 Subject: [PATCH] fix transpose cuda --- norch/tensor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/norch/tensor.py b/norch/tensor.py index 5b5cc57..03b37ee 100644 --- a/norch/tensor.py +++ b/norch/tensor.py @@ -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: