Refactor msr
Created by: APJansen
Adds a regression test for the MSR_Normalization
layer.
Mainly improving readability by rewriting the indices.
Also makes it compatible with higher rank pdfs by changing a flatten(y)
to y[0]
and allowing for an out_shape
rather than only a 1d output_dim
.
Awaiting further tests as per discussion here.