jax.numpy.isscalar#

jax.numpy.isscalar(element)[source]#

Returns True if the type of element is a scalar type.

LAX-backend implementation of numpy.isscalar().

Original docstring below.

Parameters

element (any) – Input argument, can be of any type and shape.

Returns

val – True if element is a scalar type, False if it is not.

Return type

bool