Python 如何将NaN赋给张量元素？

我想把NaN赋给张量元素

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

我有错误

TypeError:无法将非类型分配给火炬。LongTensor

。

---

我需要它来确保没有对x的某些值进行后期复杂的计算。

以下代码将所需值设置为

nan

：

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

---

请问一个一致的问题：您说您想分配

Nan

，但您的代码明确尝试分配

None

。