tf.sparse.maximum

Returns the element-wise max of two SparseTensors.

tf.sparse.maximum(
    sp_a, sp_b, name=None
)

Assumes the two SparseTensors have the same shape, i.e., no broadcasting. Example:

sp_zero = sparse_tensor.SparseTensor([[0]], [0], [7])
sp_one = sparse_tensor.SparseTensor([[1]], [1], [7])
res = tf.sparse.maximum(sp_zero, sp_one).eval()
# "res" should be equal to SparseTensor([[0], [1]], [0, 1], [7]).