tensorflow中的boolean_mask

將mask中所有為true的抽取出來，放到一起，這里從n維降到1維度

   

tensor = [[1, 2], [3, 4], [5, 6]]

import numpy as np

mask=np.array([[True,True],[False,True],[False,False]])

z=tf.boolean_mask(tensor, mask)

sess.run(z)

array([1, 2, 4])

   

   

   

tensor = [[[1, 2], [3, 4], [5, 6]],[[1, 2], [3, 4], [5, 6]]]

import numpy as np

mask=np.array([[[True,True],[False,True],[False,False]],[[True,True],[False,True],[False,False]]])

z=tf.boolean_mask(tensor, mask)

sess.run(z)

array([1, 2, 4, 1, 2, 4])