jax_equal¶
- probly.representation.jax_functions.jax_equal(x1: JaxArrayLike, x2: JaxArrayLike, /) jax.Array[source]¶
- probly.representation.jax_functions.jax_equal(x1: JaxOperand, x2: JaxOperand, /) object
Compare operands for equality, mirroring
jax.numpy.equal.- Parameters:
x1 – Left operand.
x2 – Right operand.
- Returns:
An elementwise boolean equality array for native operands, or a custom override result. Protected-axis overrides return batch-shaped masks.