AlphaFold2代码阅读（七）

最新推荐文章于 2025-10-05 16:53:31 发布

原创

最新推荐文章于 2025-10-05 16:53:31 发布 · 849 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习 #python #机器学习

2021SC@SDUSC

1.class TriangleMultiplication源码

class TriangleMultiplication(hk.Module):

  def __init__(self, config, global_config, name='triangle_multiplication'):
    super().__init__(name=name)
    self.config = config
    self.global_config = global_config

  def __call__(self, act, mask, is_training=True):
  
    del is_training
    c = self.config
    gc = self.global_config

    mask = mask[..., None]

    act = hk.LayerNorm(axis=[-1], create_scale=True, create_offset=True,
                       name='layer_norm_input')(act)
    input_act = act

    left_projection = common_modules.Linear(
        c.num_intermediate_channel,
        name='left_projection')
    left_proj_act = mask * left_projection(act)