jax.numpy.iterable

目录

jax.numpy.iterable#

jax.numpy.iterable(y)[源代码]#

检查一个对象是否可以被迭代。

参数:

y (object) – 输入对象。

返回:

b – 如果对象具有迭代器方法或是一个序列,则返回 True ,否则返回 False

返回类型:

bool

示例

>>> import numpy as np
>>> np.iterable([1, 2, 3])
True
>>> np.iterable(2)
False

备注

在大多数情况下,np.iterable(obj) 的结果与 isinstance(obj, collections.abc.Iterable) 一致。一个显著的例外是对 0 维数组的处理:

>>> from collections.abc import Iterable
>>> a = np.array(1.0)  # 0-dimensional numpy array
>>> isinstance(a, Iterable)
True
>>> np.iterable(a)
False