我想将NaN分配给张量元素。
import torch
x = torch.tensor([1, 2, 3])
x[x == 2] = None
我有错误TypeError: can't assign a NoneType to a torch.LongTensor
。我需要它来确保不对某些x值进行某些稍后的复杂计算。
以下代码将所需的值设置为nan
:
import torch
x = torch.tensor([1, 2, 3]).float()
x[x == 2] = float('nan')
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句