This bijector is identical to the "Convolution1x1" used in Glow (Kingma and Dhariwal, 2018).

tfb_scale_matvec_lu(
  lower_upper,
  permutation,
  validate_args = FALSE,
  name = NULL
)

Arguments

lower_upper

The LU factorization as returned by tf$linalg$lu.

permutation

The LU factorization permutation as returned by tf$linalg$lu.

validate_args

Logical, default FALSE. Whether to validate input with asserts. If validate_args is FALSE, and the inputs are invalid, correct behavior is not guaranteed.

name

name prefixed to Ops created by this class.

Value

a bijector instance.

References

See also