jax.numpy.isscalar

jax.numpy.isscalar(element)

如果 element 的类型是标量类型，则返回 True。

numpy.isscalar() 的 LAX 后端实现。

原始文档字符串如下。

参数：

element (any) – 输入参数，可以是任何类型和形状。

返回值：

val – 如果 element 是标量类型，则为 True，否则为 False。

返回类型：

bool