sortie 2d valeur masquée par boolean tensorflow

Question

j'ai une entrée [n, m] et [n, m] masque booléenne. Comment pourrais-je produire une matrice filtrée [n, x], au lieu de [x] tableau?
Par exemple, l'entrée

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

et un masque booléen

bm = [[1,0,1],[1,0,0]] 

J'ai essayé d'utiliser tf.boolean_mask() et je suis arrivé [1,3,4]. Comment pourrais-je obtenir un résultat 2D comme

result = [[1,3],[4]] 

Merci!

Answer 1

Cette opération ne peut exister dans TF car TF fonctionne sur les tenseurs et retourne tenseurs. Le résultat que vous avez reçu a un nombre différent d'éléments et non un tenseur.

Answer 2

Il n'y a aucun moyen de le faire dans TF. Cependant, vous pouvez utiliser Masked Array en numpy. Vous pouvez faire quelque chose comme

>>> import numpy as np, numpy.ma as ma 
>>> x = ma.array([1., -1., 3., 4., 5., 6.], mask=[0,0,0,0,1,0]) 
>>> y = ma.array([1., 2., 0., 4., 5., 6.], mask=[0,0,0,0,0,1]) 
>>> print np.sqrt(x/y) 
[1.0 -- -- 1.0 -- --]