Triton 是一種用于并行編程的語言和編譯器。它旨在提供一個基于 Python 的編程環境，以高效編寫自定義 DNN 計算內核，并能夠在現代 GPU 硬件上以最大吞吐量運行。

更多 Triton 中文文檔可訪問 →https://triton.hyper.ai/

triton.language.broadcast_to(input, *shape)

嘗試將給定的張量廣播到新的?shape。

參數：

• input (Block) - 輸入張量。
• shape - 所需的形狀。

shape?可以以 1 個元組或獨立參數被傳入：

# These are equivalent 
# 這些是等效的
broadcast_to(x, (32, 32))
broadcast_to(x, 32, 32)

該函數也可作為?tensor?的 1 個成員函數調用，使用?x.broadcast_to(...)?而不是?broadcast_to(x, ...)。