Searched defs:AxisLike (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/_numpy/ |
| D | _reductions_impl.py | 67 def count_nonzero(a: ArrayLike, axis: AxisLike = None, *, keepdims: KeepDims = False):
74 axis: AxisLike = None,
94 axis: AxisLike = None,
114 axis: AxisLike = None,
128 axis: AxisLike = None,
142 axis: AxisLike = None,
160 axis: AxisLike = None,
178 axis: AxisLike = None,
188 axis: AxisLike = None,
207 axis: AxisLike = None,
[all …]
|
| D | _normalizations.py | 22 AxisLike = typing.TypeVar("AxisLike") variable
|