Skip to main content

chain_pair_features

ferritin_plms::esmfold2::layers::pair_init

Function chain_pair_features 

Source 
pub fn chain_pair_features(chain_ids: &Tensor) -> Result<Tensor>
Expand description

Per-pair same-chain / different-chain indicator.

§Arguments

  • chain_ids — [B, N] integer chain identifiers (0, 1, 2, …)

§Returns

[B, N, N, 2] float32, where dim-3 is [same_chain, diff_chain].