AlphaFold2代码阅读（五）

最新推荐文章于 2025-11-13 10:06:35 发布

原创

最新推荐文章于 2025-11-13 10:06:35 发布 · 951 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#python

2021SC@SDUSC

class TemplatePairStack

class TemplatePairStack(hk.Module)

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

  def __call__(self, pair_act, pair_mask, is_training, safe_key=None):

class TemplatePairStack负责的是模板的配对堆栈
参数：
pair_act：单个模板的配对激活，形状为 [N_res, N_res, c_t]。
pair_mask：对掩码，形状为 [N_res, N_res]。
is_training：模块是否处于训练模式。
safe_key：封装随机数生成密钥的安全密钥对