如何将 NaN 赋给张量元素？

Question

我想将 NaN 分配给张量元素。

import torch
x = torch.tensor([1, 2, 3])
x[x == 2] = None

我有错误TypeError: can't assign a NoneType to a torch.LongTensor。我需要它来确保不会对 x 的某些值进行一些后续的复杂计算。

Answer 1

以下代码将设置所需的值nan：

import torch
x = torch.tensor([1, 2, 3]).float()
x[x == 2] = float('nan')