Loading

2022年5月8日
优雅地实现多头自注意力——使用einsum（爱因斯坦求和）进行矩阵运算
摘要： einsum函数说明 pytorch文档说明：$torch.einsum(equation, **operands)$ 使用基于爱因斯坦求和约定的符号，将输入operands的元素沿指定的维数求和。einsum允许计算许多常见的多维线性代数阵列运算，方法是基于爱因斯坦求和约定以简写格式表示它们。主要 阅读全文
posted @ 2022-05-08 11:50 MapleTx 阅读(3680) 评论(0) 推荐(0) 编辑