欢迎来到 PyTorch 教程¶
PyTorch 教程的新内容?
通过用嵌套张量和 torch.compile() 替换 nn.Transformer 来加速 PyTorch Transformer
更新了 torch.export 教程,具有自动动态形状
Dim.AUTO
更新了 将用户定义的 Triton 内核与 torch.compile 结合使用,具有新的
torch.library.triton_op
PyTorch 教程的新内容?
通过用嵌套张量和 torch.compile() 替换 nn.Transformer 来加速 PyTorch Transformer
更新了 torch.export 教程,具有自动动态形状 Dim.AUTO
更新了 将用户定义的 Triton 内核与 torch.compile 结合使用,具有新的 torch.library.triton_op