Glow: The Affine Coupling Layer
Advanced
The problem, from first principles
Glow's affine coupling layer splits the features in two. One half passes through unchanged and is used to compute a scale and shift for the other half. Because the first half is known, the transform can be inverted exactly.
From theory to code
Implement affine_coupling_forward(xa, xb, s, t), returning the unchanged half and the scaled and shifted half.
Constraints
sis a log-scale, so the scale factor isexp(s).
Hints
Hint 1
Keep xa as is, and compute xb * exp(s) + t.
Keywords
research-papersunsupervisednormalizing-flowsgenerative-models
Python
Run your code to see results
No results yet
Run your code to check it against the examples, or submit to run every test.