pygmtools.utils.inner_prod_aff_fn
- pygmtools.utils.inner_prod_aff_fn(feat1, feat2, backend=None)[source]
Inner product affinity function. The affinity is defined as
\[\mathbf{f}_1^\top \cdot \mathbf{f}_2\]- Parameters
feat1 – \((b\times n_1 \times f)\) the feature vectors \(\mathbf{f}_1\)
feat2 – \((b\times n_2 \times f)\) the feature vectors \(\mathbf{f}_2\)
backend – (default:
pygmtools.BACKEND
variable) the backend for computation.
- Returns
\((b\times n_1\times n_2)\) element-wise inner product affinity matrix