Searched refs:array_or_scalar (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_numpy/ |
| H A D | random.py | 19 from ._normalizations import array_or_scalar, ArrayLike, normalizer
88 return array_or_scalar(values, return_scalar=size == ())
107 return array_or_scalar(values, return_scalar=size == ())
114 return array_or_scalar(values, return_scalar=size == ())
123 return array_or_scalar(values, return_scalar=size == ())
152 return array_or_scalar(values, int, return_scalar=size == ())
|
| H A D | _normalizations.py | 191 def array_or_scalar(values, py_type=float, return_scalar=False): function
|