TensorFlow數(shù)學(xué)函數(shù)：tf.setdiff1d

tf.setdiff1d 函數(shù)

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

定義在：tensorflow/python/ops/array_ops.py.

請參閱指南：數(shù)學(xué)>序列比較和索引,張量轉(zhuǎn)換>分割和連接

計算兩個數(shù)字或字符串列表之間的差異.

給定一個列表x和一個列表y,這個操作返回一個列表out,它表示 x 而不是 y 中的所有值.返回的列表out按照數(shù)字出現(xiàn)的順序排序x(保留重復(fù)項).該操作還返回一個列表idx表示每個x中out元素的位置.換一種說法：

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

例如,給出這個輸入：

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

該操作將返回：

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

函數(shù)參數(shù)

• x：一個Tensor；1-d；要保持的值.
• y：一個Tensor；必須與x具有相同的類型；1-d；要刪除的值.
• out_idx：可選的tf.DType來自：tf.int32, tf.int64；默認(rèn)為tf.int32.
• name：操作的名稱(可選).

函數(shù)返回值

Tensor對象的元組(out,idx).

• out：一個Tensor.與x有相同的類型；1-d；出現(xiàn)在x,但而不是y中的值.
• idx：out_idx類型的Tensor；1-d；x的值的位置保存在out.