IXThinker·2026-01-19 18:50Triton教程：where算子的條件選擇用法
Triton教程系列介紹triton.language.where的用法：按布爾條件張量逐元素在x與y之間選值。看似簡單的算子，其實有求值與掩碼使用的細節值得留意。
基本語義解析
where(condition, x, y)會根據條件張量逐元素取值：條件為真時取x，否則取y；x與y的形狀會被廣播至條件的形狀，兩者必須具有相同的數據類型。
IXThinker 會員內容
呢篇係 Archive 內容。登入即可解鎖全文，仲有成個 15,000+ 篇、每日更新嘅 AI 動態資料庫。
免費登入／註冊解鎖全文 →Google 或電郵地址即可，即刻解鎖。