去除所有的nan并替换成0

解决方法

    angle = torch.where(torch.isnan(angle), torch.full_like(angle, 0), angle)
    print(torch.any(torch.isnan(angle)))

---

torch.where(condition, x, y)

• 参数1；判断条件
• 参数2：满足条件，返回x
• 参数3：满足条件，返回y
• out: 返回修改的数据

---

torch.isnan(angle)

• 参数angle: 输入
• out: 为nan的返回True, 不是的返回Flase

---

torch.full_like(angle, 0)

• 参数angle: 输入
• 参数2： 替换成的元素