在 Tensorflow 中使用 2D 占位符屏蔽 3D 占位符张量

Question

我有一个动态大小为 A（批量大小，序列大小，5）的占位符和另一个具有零值和一个值和动态大小 B（批量大小，序列大小）的占位符。我想使用第二个 2D 占位符来屏蔽第一个占位符，就好像张量 B[0][0]=0 的值那么 A[0][0][0:5] 将被设置为零，如果它是等于 B[0][0]=1 则 A[0][0][0:5] 不会改变。

palceholder A:(shape=(2,2,5))    

   [[[ 1,  2,  3,  1,  4],

    [ 2,  3,  5,  2,  4]],

   [[ 2,  7,  5,  8,  1],

    [ 4,  5,  1,  3,  9]]] 

palceholder B:(shape=(2,2))

  [[ 1,  0],

   [ 0,  1]]

Tensor C= Mask(A,B)    

   [[[ 1,  2,  3,  1,  4],

    [ 0,  0,  0,  0,  0]],

   [[ 0,  0,  0,  0,  0],

    [ 4,  5,  1,  3,  9]]]

我试过 tf.boolean_mask 但它不适用于动态大小的掩码。