Follows implementation of RIM of paper "Recurrent Inference Machines for Solving Inverse Problems" (https://arxiv.org/pdf/1706.04008.pdf).
Follows implementation of RIM of paper "Recurrent Inference Machines for Solving Inverse Problems" (https://arxiv.org/pdf/1706.04008.pdf).