Non-square images can be exported to torch tensors