torch.Tensor.isclose

Tensor.isclose(other, rtol=1e-05, atol=1e-08, equal_nan=False) → Tensor

请参阅 torch.isclose()