Skip to content

Commit 186be13

Browse files
authored
Merge pull request #30 from Quansight-Labs/byte_strides
make ndarray.strides byte strides to match numpy
2 parents 085fa8f + ace30bf commit 186be13

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
@@ -45,7 +45,8 @@ def dtype(self):
4545

4646
@property
4747
def strides(self):
48-
return self._tensor.stride() # XXX: byte strides
48+
elsize = self._tensor.element_size()
49+
return tuple(stride * elsize for stride in self._tensor.stride())
4950

5051
@property
5152
def base(self):

0 commit comments

Comments
 (0)