Triton教程:flip函數翻轉張量用法
Triton中文文檔介紹triton.language.flip函數的用法:該函數沿指定維度翻轉張量,目前僅支持最後一維,參數設計簡潔,並支持以成員函數形式調用。
語言定位
Triton是一種用於並行編程的語言與編譯器,提供基於Python的編程環境,專為高效編寫自定義DNN計算內核而設計,可在現代GPU上以最大吞吐量運行。
IXThinker 會員內容
呢篇係 Archive 內容。登入即可解鎖全文,仲有成個 15,000+ 篇、每日更新嘅 AI 動態資料庫。
免費登入/註冊解鎖全文 →Google 或電郵地址即可,即刻解鎖。
