I just got started with complex tensors and tried assigning a complex tensor of shape (2,2) and it runs out of memory. My code is -
import torch import torch.nn as nn a = torch.rand(size=(2,2), dtype=torch.complex32) print(a)
and the output error is -
Traceback (most recent call last): File "/home/atharva/PycharmProjects/DeepComplexNetwork/main.py", line 4, in <module> a = torch.rand(size=(2,2), dtype=torch.complex32) RuntimeError: [enforce fail at CPUAllocator.cpp:64] . DefaultCPUAllocator: can't allocate memory: you tried to allocate 34359738368 bytes. Error code 12 (Cannot allocate memory)
I was just wondering if complex tensor of type
complex32 even of size (2,2)
34359738368 bytes big ?