我有一個動態大小為 A(批量大小,序列大小,5)的占位符和另一個具有零值和一個值和動態大小 B(批量大小,序列大?。┑恼嘉环N蚁胧褂玫诙€ 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 但它不適用于動態大小的掩碼。
添加回答
舉報
0/150
提交
取消