Coding Exercise on Barlow Twins

The Barlow Twins objective function

Given the representations of distorted views of a batch, $Z_1$ and $Z_2$, and assuming they are mean-centered (zero mean and unit variance) along the batch, the Barlow Twins algorithm first computes a correlation matrix $\mathcal{C}$ ...