BooleanMask

ใช้เพื่อระบุแกนที่จะมาสก์ axis + dim(mask) <= dim(tensor) และรูปร่างของ mask ต้องตรงกับขนาด axis + dim(mask) ของรูปร่างของ tensor

ใช้เพื่อระบุแกนที่จะมาสก์ axis + dim(mask) <= dim(tensor) และรูปร่างของ mask ต้องตรงกับขนาด axis + dim(mask) ของรูปร่างของ tensor

ใช้มาสก์บูลีนกับเทนเซอร์ ส่งกลับอาร์เรย์แบบแบนของแต่ละองค์ประกอบที่สอดคล้องกับ true ในมาสก์

เทียบเท่ากับ Numpy คือ tensor[mask]

โดยทั่วไป 0 < dim(mask) = K <= dim(tensor) และรูปร่างของ mask จะต้องตรงกับมิติ K แรกของรูปร่างของ tensor จากนั้นเราก็จะได้: booleanMask(tensor, mask)[i, j1,...,jd] = tensor[i1,...,iK,j1,...,jd] โดยที่ (i1,...,iK) เป็นการเข้ามา true ของ mask (ลำดับหลักแถว)

axis สามารถใช้กับมา mask เพื่อระบุแกนที่จะมาสก์ได้ (โดยค่าเริ่มต้นจะเป็น 0) ในกรณีนั้น axis + dim(mask) <= dim(tensor) และ mask จะต้องตรงกับขนาด axis + dim(mask) ของรูปร่างของ tensor