↓ 7 callersFunctioncollate_dense_tensorsTakes a list of tensors with the following dimensions: [(d_11, ..., d_1K), (d_21, ..., d_2K), ..., (d_N1,
src/byprot/models/structok/modules/folding_utils/misc.py:159
↓ 6 callersFunctiontopk_masking scores: [b, n] cutoff_len: [b, 1] stochastic: bool, whether to add noise to select top_k or not returns: mask: [b, n], with 1
src/byprot/models/utils.py:257
↓ 4 callersFunctionNWDP_TMInput: score[1:len1, 1:len2], and gap_open
* Output: j2i[1:len2] \in {1:len1} U {-1}
* path[0:len1, 0:len2]=1,2,3, from diagonal, horizontal, vert
analysis/TMalign.cpp:1321