diffalign.models.common.extend_to_cross_attention

diffalign.models.common.extend_to_cross_attention(pos, cutoff, batch, graph_idx)