Skip to content

Commit ace30bf

Browse files
committed
API: make ndarray.strides byte strides to match numpy
1 parent 69888d7 commit ace30bf

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

torch_np/_ndarray.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,8 @@ def dtype(self):
7070

7171
@property
7272
def strides(self):
73-
return self._tensor.stride() # XXX: byte strides
73+
elsize = self._tensor.element_size()
74+
return tuple(stride * elsize for stride in self._tensor.stride())
7475

7576
@property
7677
def base(self):

0 commit comments

Comments
 (0)