From 06103c8266e6be1551586d8c099adce935264e64 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Thu, 24 Oct 2019 16:26:59 +0200
Subject: [PATCH] Bugfix: torch.stride is strides in element_size()
---
src/pystencils_autodiff/field_tensor_conversion.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/pystencils_autodiff/field_tensor_conversion.py b/src/pystencils_autodiff/field_tensor_conversion.py
index af38996..ca0ae54 100644
--- a/src/pystencils_autodiff/field_tensor_conversion.py
+++ b/src/pystencils_autodiff/field_tensor_conversion.py
@@ -30,7 +30,7 @@ def _torch_tensor_to_numpy_shim(tensor):
from pystencils.autodiff.backends._pytorch import torch_dtype_to_numpy
fake_array = _WhatEverClass(
- strides=[tensor.stride(i) for i in range(len(tensor.shape))],
+ strides=[tensor.stride(i) * tensor.storage().element_size() for i in range(len(tensor.shape))],
shape=tensor.shape,
dtype=torch_dtype_to_numpy(tensor.dtype))
return fake_array
--
GitLab