Where3

بسته به «شرایط»، عناصر را از «x» یا «y» انتخاب می‌کند.

تانسورهای 'x' و 'y' همگی باید یک شکل باشند و خروجی نیز آن شکل را خواهد داشت.

اگر «x» و «y» اسکالر هستند، تانسور شرط باید یک اسکالر باشد. اگر «x» و «y» بردار یا رتبه بالاتری باشند، «شرط» باید یا یک اسکالر باشد، یا بردار با اندازه ای که با بعد اول «x» مطابقت داشته باشد، یا باید همان شکل «x» باشد.

تانسور «شرط» به‌عنوان یک ماسک عمل می‌کند که بر اساس مقدار هر عنصر، انتخاب می‌کند که آیا عنصر / ردیف مربوطه در خروجی باید از «x» (در صورت درست) یا «y» (اگر نادرست) گرفته شود.

اگر «شرط» یک بردار باشد و «x» و «y» ماتریس‌های رتبه بالاتری هستند، آنگاه انتخاب می‌کند که کدام ردیف (بعد بیرونی) را از «x» و «y» کپی کند. اگر «شرط» شکلی مشابه «x» و «y» داشته باشد، انتخاب می‌کند کدام عنصر را از «x» و «y» کپی کند.

به عنوان مثال:

# 'condition' tensor is [[True,  False]
 #                        [False, True]]
 # 't' is [[1, 2],
 #         [3, 4]]
 # 'e' is [[5, 6],
 #         [7, 8]]
 select(condition, t, e)  # => [[1, 6], [7, 4]]
 
 
 # 'condition' tensor is [True, False]
 # 't' is [[1, 2],
 #         [3, 4]]
 # 'e' is [[5, 6],
 #         [7, 8]]
 select(condition, t, e) ==> [[1, 2],
                              [7, 8]]