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