ai-content-maker/.venv/Lib/site-packages/torch/_numpy/_getlimits.py

16 lines
269 B
Python

# 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)