Блог

Главная - Вопросы по программированию - Как создать torch.tensor с формой (1,1,32) со значением по умолчанию?

Как создать torch.tensor с формой (1,1,32) со значением по умолчанию?

#pytorch

#pytorch

Вопрос:

Я хочу создать torch.tensor переменную с формой (1,1,32) со значением по умолчанию ( None ).

Как я могу это сделать?

Ответ №1:

Я не верю, что вы можете назначить None a torch.Tensor . Однако более уместно вместо этого назначить NaN . Вы можете сделать это с помощью встроенного 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]]])
Метки: , ,