Spaces:
Running
Running
File size: 284 Bytes
c61ccee |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
# mypy: ignore-errors
import torch
from . import _dtypes
def finfo(dtyp):
torch_dtype = _dtypes.dtype(dtyp).torch_dtype
return torch.finfo(torch_dtype)
def iinfo(dtyp):
torch_dtype = _dtypes.dtype(dtyp).torch_dtype
return torch.iinfo(torch_dtype)
|