jax.numpy.isscalar

jax.numpy.isscalar(element)[source]

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

LAX-backend implementation of 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