在 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 但它不适用于动态大小的掩码。