Algorithm 1: Pseudocode of PIC in a PyTorch-likestyle