小编pai*_*ood的帖子

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

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

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