Possible bug in SimpleITK or in PyTorch

SITK_PT.ipynb

The code, just in case:

# -*- coding: utf-8 -*-
"""Untitled8.ipynb

Automatically generated by Colaboratory.

Original file is located at
    https://colab.research.google.com/drive/1GzO2RQYn7jI26frWS8S-YlbSmjFAehWa
"""

# pip install --quiet torchio==0.17.48

import torch
import torchio as tio
import SimpleITK as sitk

transform = tio.RandomAffine()
ixi = tio.datasets.IXITiny('ixi_tiny', download=True, transform=transform)

paths = [ixi[i].image.path for i in range(4)]
print(paths)


class MyDataset(torch.utils.data.Dataset):
    def __init__(self, paths):
        self.paths = paths

    def __len__(self):
        return len(self.paths)

    def __getitem__(self, index):
        path = self.paths[index]
        image = sitk.ReadImage(str(path))

        resampler = sitk.ResampleImageFilter()
        resampler.SetInterpolator(sitk.sitkNearestNeighbor)
        resampler.SetReferenceImage(image)
        resampler.SetDefaultPixelValue(0.0)
        resampler.SetOutputPixelType(sitk.sitkFloat32)
        resampled = resampler.Execute(image)

        return sitk.GetArrayFromImage(resampled)

my_dataset = MyDataset(paths)

loader_sp = torch.utils.data.DataLoader(my_dataset, batch_size=4, num_workers=0)
loader_mp = torch.utils.data.DataLoader(my_dataset, batch_size=4, num_workers=2)

batch_sp = next(iter(loader_sp))
print(batch_sp.shape)

batch_mp = next(iter(loader_mp))
print(batch_mp.shape)