mindsponge.cell

接口名

概述

支持平台

mindsponge.cell.Attention

多头注意力机制,具体实现请参考 Attention is all you need

Ascend GPU

mindsponge.cell.GlobalAttention

global gated自注意力机制,具体实现请参考 Highly accurate protein structure prediction with AlphaFold

Ascend GPU

mindsponge.cell.InvariantPointAttention

该模块用于更新序列表示(即输入inputs_1d),在序列表示中加入位置信息。

Ascend GPU

mindsponge.cell.MSAColumnAttention

MSA列注意力层。

Ascend GPU

mindsponge.cell.MSAColumnGlobalAttention

MSA列全局注意力层。

Ascend GPU

mindsponge.cell.MSARowAttentionWithPairBias

MSA行注意力层。

Ascend GPU

mindsponge.cell.OuterProductMean

通过外积平均计算输入张量(act)在第二维上的相关性,得到的相关性可以用于更新相关特征(如Pair特征)。

Ascend GPU

mindsponge.cell.Transition

两层全连接层,中间输出特征数为输入特征数的倍数。

Ascend GPU

mindsponge.cell.TriangleAttention

三角注意力机制。

Ascend GPU

mindsponge.cell.TriangleMultiplication

三角乘法层。

Ascend GPU