torch.isnan¶ torch.isnan(input) → Tensor¶ 返回一个新的张量,其中布尔元素表示 input 的每个元素是否为 NaN。当复数的实部和/或虚部为 NaN 时,则认为该复数为 NaN。 参数 input (Tensor) – 输入张量。 返回值 一个布尔张量,在 input 为 NaN 的位置为 True,其他位置为 False 示例 >>> torch.isnan(torch.tensor([1, float('nan'), 2])) tensor([False, True, False])