So the issue is just that your don’t have enough memory to store that Tensor. I’m afraid you will need to find a way with not representing the full Tensor, or having a smaller one. Or buy a gpu with more memory.