pytorch

1
# mypy: ignore-errors
2

3
import torch
4

5
from . import _dtypes
6

7

8
def finfo(dtyp):
9
    torch_dtype = _dtypes.dtype(dtyp).torch_dtype
10
    return torch.finfo(torch_dtype)
11

12

13
def iinfo(dtyp):
14
    torch_dtype = _dtypes.dtype(dtyp).torch_dtype
15
    return torch.iinfo(torch_dtype)
16