我有一個動態(tài)大小為 A（批量大小，序列大小，5）的占位符和另一個具有零值和一個值和動態(tài)大小 B（批量大小，序列大?。┑恼嘉环?。我想使用第二個 2D 占位符來屏蔽第一個占位符，就好像張量 B[0][0]=0 的值那么 A[0][0][0:5] 將被設(shè)置為零，如果它是等于 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 但它不適用于動態(tài)大小的掩碼。