The expanding of the tensors did not happen.
    print(predicted_model[0].shape)
    print(voxel_array.shape)

still gives
torch.Size([1, 384, 384, 384])
torch.Size([1, 384, 384, 384])

so i guess i’ll have too figure out why they are not transformed…