Logical operations
./code/logical-op.py
1#!/usr/bin/env python3
2
3import torch
4
5a = torch.tensor([float("inf")])
6b = torch.tensor([float("nan")])
7assert torch.isinf(a).item() is True
8assert torch.isnan(a).item() is False
9
10assert torch.isinf(b).item() is False
11assert torch.isnan(b).item() is True
12
13assert torch.logical_or(torch.isinf(a), torch.isnan(b)).item() is True
14
15assert a.isinf().item() is True
16assert a.isnan().item() is False
17
18assert b.isinf().item() is False
19assert b.isnan().item() is True