[FIXED] How to create torch.tensor with shape (1,1,32) with default value?

Issue

I want to create torch.tensor variable with shape (1,1,32) with default value (None).

How can I do it ?

Solution

I don't believe you can assign None to a torch.Tensor. What is more appropriate however is to instead assign NaN. You can do so using the builtin torch.full:

>>> torch.full((1, 1, 32), torch.nan)
tensor([[[nan, nan, nan, nan, nan, nan, nan, nan, 
          nan, nan, nan, nan, nan, nan, nan, nan, 
          nan, nan, nan, nan, nan, nan, nan, nan, 
          nan, nan, nan, nan, nan, nan, nan, nan]]])

Answered By - Ivan