【Triton 教程】triton_language.broadcast_to
Triton是一种用于并行编程的语言和编译器。它旨在提供一个基于Python的编程环境,以高效编写自定义DNN计算内核,并能够在现代GPU硬件上以最大吞吐量运行。更多Triton中文文档可访问→triton.language.broadcastto(input,shape)尝试将给定的张量广播到新的shape。参数:input(Block)输入张量。shape所需的形状。shape可以以1个元组或独立参数被传入:Theseareequivalent这些是等效的broadcastto(x,(32,32))broadcastto(x,32,32)...