jax.numpy.iterable

内容

jax.numpy.iterable#

jax.numpy.iterable(y)[source]#

检查对象是否可以迭代。

参数:

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