tf.boolean_mask就是true 對應(yīng)位置的張量

import tensorflow as tf import numpy as np # 1-D example tensor = [0, 1, 2, 3] mask = np.array([True, False, True, False]) c=tf.boolean_mask(tensor, mask) # [0, 2]print(tf.keras.backend.eval(c)) [0 2] # 2-D example tensor = [[1, 2], [3, 4], [5, 6]] mask = np.array([True, False, True]) c=tf.boolean_mask(tensor, mask) # [[1, 2], [5, 6]]print(tf.keras.backend.eval(c)) [[1 2][5 6]]

總結(jié)

以上是生活随笔為你收集整理的tf.boolean_mask的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。