tf.boolean_mask 的作用是 通過(guò)布爾值 過(guò)濾元素

def boolean_mask(tensor, mask, name="boolean_mask", axis=None):"""Apply boolean mask to tensor."""

參數(shù)解釋：

tensor：被過(guò)濾的元素列表或數(shù)組

mask：一堆 bool 值，它的維度不一定等于 tensor

return： mask 為 true 對(duì)應(yīng)的 tensor 的元素

當(dāng) tensor 與 mask 維度一致時(shí)，return 一維

# 1維的示例 tensor = [0, 1, 2, 3] mask = np.array([True, False, True, False]) out = tf.boolean_mask(tensor, mask) with tf.Session() as sess:print(sess.run(out)) # [0, 2]print(out.shape) # (?,)

再看看 mask 與 tensor 維度不同的例子

tensor = [[1, 2], [3, 4], [5, 6]] mask = np.array([True, False, True]) # mask 與 tensor 維度不同 out2 = tf.boolean_mask(tensor, mask) with tf.Session() as sess:print(sess.run(out2)) # [[1, 2], [5, 6]]print(out2.shape) # (?, 2)

mask 可以用一個(gè)函數(shù)代替

# 3-D tensor = tf.constant([[[2,4],[4,1]],[[6,8],[2,1]]],tf.float32) mask = tensor > 2 # 濾波器 mask 與 tensor 相同維度 out3 = tf.boolean_mask(tensor, mask) with tf.Session() as sess:print(sess.run(tensor))print(sess.run(mask)) # [[[False True] [ True False]]# [[ True True] [False False]]]print(sess.run(out3)) # [4. 4. 6. 8.] 輸出一維print(out3.shape) # (?,)

?

總結(jié)

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

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