zhusuan.transforms.invertible.scaling

class Scaling(n_dim)

Bases: zhusuan.transforms.invertible.base.InvertibleTransform

The scaling layer described in NICE paper [DKB15], which compute the following process and its inverse.

\[ \begin{align}\begin{aligned}\begin{bmatrix} S_1 & & & \\ & S_2 & & \\ & & \ddots & \\ & & & S_D \end{bmatrix} \begin{bmatrix} h_{i - 1, 1} \\h_{i - 1, 2} \\\vdots \\h_{i - 1, D} \end{bmatrix} = \begin{bmatrix} h_{i, 1} \\h_{i , 2} \\\vdots \\h_{i, D} \end{bmatrix}\end{aligned}\end{align} \]

Parameters

n_dim – The dim of the Var to be transformed.

References

DKB15

Laurent Dinh, David Krueger, and Yoshua Bengio. Nice: non-linear independent components estimation. 2015. arXiv:1410.8516.