I’m trying to subtract two tensors, which I don’t think should be prohibitively large…
uv.shape = torch.Size([256, 256, 2])
xy.shape = torch.Size([40, 28672, 3])
When I try to perform a simple subtraction, my kernel restarts by itself. I’m guessing it’s a memory issue, but don’t see why my computer can’t handle it as I have 128GB RAM. What could be triggering this problem?
uv[None,None,:,:,0]-xy[:, :,None,None,0]