jax.numpy.floor

Contents

jax.numpy.floor#

jax.numpy.floor(x, /)[source]#

Return the floor of the input, element-wise.

LAX-backend implementation of numpy.floor().

Original docstring below.

The floor of the scalar x is the largest integer i, such that i <= x. It is often denoted as \(\lfloor x \rfloor\).

Parameters:

x (array_like) – Input data.

Returns:

y – The floor of each element in x. This is a scalar if x is a scalar.

Return type:

ndarray or scalar