Google Git
Sign in
android / platform / external / pytorch / b5dd37f23efecdd27d1ff8862dcd59613c36211d / . / torch / _numpy / _getlimits.py
blob: dc2cc845f752917caf2d8817f4f28be2b56f75e3 [file] [log] [blame]
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)