tf.compat.v1.setdiff1d  |  TensorFlow v2.16.1 (original) (raw)

tf.compat.v1.setdiff1d

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

Computes the difference between two lists of numbers or strings.

tf.compat.v1.setdiff1d(
    x,
    y,
    index_dtype=tf.dtypes.int32,
    name=None
)

Given a list x and a list y, this operation returns a list out that represents all values that are in x but not in y. The returned list outis sorted in the same order that the numbers appear in x (duplicates are preserved). This operation also returns a list idx that represents the position of each out element in x. In other words:

out[i] = x[idx[i]] for i in [0, 1, ..., len(out) - 1]

For example, given this input:

x = [1, 2, 3, 4, 5, 6]
y = [1, 3, 5]

This operation would return:

out ==> [2, 4, 6]
idx ==> [1, 3, 5]