انتخاب عناصر تنسور با تنسور boolean

Question

سلام.

تو پایتون اگر یک لیست boolean داشته باشیم می تونیم از لیست دیگر به صورت نظیر به نظیر عناصری که true هستند را مقادیرشان را داشته باشیم

list_bool = [True,False,True,False,False]
list = [1,2,3,4,5]
result = list[list_bool]


result : [1,3]

معادل همین رو توی tensorflow نیاز دارم.

Answer 1

از تابع boolean_mask استفاده کنید پارامتر تنسور داده و پارامتر تنسور ماسک را پاس بدید توجه داشته باشید که تنسور ماسک باید از نوع boolean باشه