Attend the Women in ML Symposium on December 7 Register now

tfp.experimental.distributions.marginal_fns.ps.equal

Stay organized with collections Save and categorize content based on your preferences.

Returns the truth value of (x == y) element-wise.

Performs a broadcast with the arguments and then an element-wise equality comparison, returning a Tensor of boolean values.

For example:

x = tf.constant([2, 4])
y = tf.constant(2)
tf.math.equal(x, y)
<tf.Tensor: shape=(2,), dtype=bool, numpy=array([ True,  False])>
x = tf.constant([2, 4])
y = tf.constant([2, 4])
tf.math.equal(x, y)
<tf.Tensor: shape=(2,), dtype=bool, numpy=array([ True,  True])>

x A tf.Tensor.
y A tf.Tensor.
name A name for the operation (optional).

A tf.Tensor of type bool with the same size as that of x or y.

tf.errors.InvalidArgumentError: If shapes of arguments are incompatible