jax.lax.sub# jax.lax.sub(x, y)[source]# Elementwise subtraction: \(x - y\). Parameters: x (Union[Array, ndarray, bool_, number, bool, int, float, complex]) y (Union[Array, ndarray, bool_, number, bool, int, float, complex]) Return type: Array