快捷键

torch.nn.attention.bias

定义与缩放点积注意力一起工作的偏置子类

因果偏差

CausalBias

表示因果注意模式的偏差。

causal_lower_right

创建一个右下角的因果偏差。

causal_upper_left

创建一个左上角的因果偏差。

CausalVariant

用于注意力机制的因果变体枚举。


© 版权所有 PyTorch 贡献者。

使用 Sphinx 构建,主题由 Read the Docs 提供。

文档

PyTorch 开发者文档全面访问

查看文档

教程

获取初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得您的疑问解答

查看资源