Contents:
ActivationNormalization
ActivationNormalization._PositiveConstraint_
ActivationNormalization._location_
ActivationNormalization._prepare_variables_for_computation_()
ActivationNormalization._scale_
ActivationNormalization.call()
ActivationNormalization.compute_jacobian_determinant()
ActivationNormalization.invert()
AdditiveCoupling
AdditiveCoupling._couple_()
AdditiveCoupling._decouple_()
AdditiveCoupling.compute_jacobian_determinant()
CheckerBoardPermutation
CheckerBoardPermutation.from_config()
CheckerBoardPermutation.generate_rope_indices()
CheckerBoardPermutation.is_end_of_axis()
Coupling
Coupling._compute_coupling_parameters_
Coupling._couple_()
Coupling._decouple_()
Coupling._mask_
Coupling.build()
Coupling.call()
Coupling.from_config()
Coupling.get_config()
Coupling.invert()
DisentanglingFlowModel
DisentanglingFlowModel.train_step()
FlowLayer
FlowLayer._axes_
FlowLayer._shape_
FlowLayer.call()
FlowLayer.get_config()
FlowLayer.invert()
FlowModel
FlowModel.add()
FlowModel.build()
FlowModel.call()
FlowModel.from_config()
FlowModel.get_config()
FlowModel.invert()
HeavisidePermutation
HeavisidePermutation.from_config()
Permutation
Permutation._forward_permutation_
Permutation._inverse_permutation_
Permutation.call()
Permutation.compute_jacobian_determinant()
Permutation.get_config()
Permutation.invert()
Reflection
Reflection._inverse_mode_
Reflection._reflect_()
Reflection._reflection_normals_
Reflection.call()
Reflection.compute_jacobian_determinant()
Reflection.get_config()
Reflection.invert()
ShufflePermutation
ShufflePermutation.from_config()
SupervisedFactorLoss
SupervisedFactorLoss.call()
CheckerBoardMask
CheckerBoardMask.from_config()
HeavisideMask
HeavisideMask.from_config()
Mask
Mask._axes_
Mask._mask_
Mask._shape_
Mask.arrange()
Mask.call()
Mask.from_config()
Mask.get_config()
Mask.re_arrange()
add_noise()
create_data_set()
factorized_pair_iterator()
reset_random_number_generators()
archimedian_spiral()
cartesian_to_polar()
logarithmic_spiral()
normal()
polar_to_cartesian()
rotate()
tangent()
cross_validate()
color_palette
evaluate_and_plot_networks()
make_2_dimensional_gaussian()
make_color_wheel()
make_radial_line()
plot_contribution_per_layer()
plot_input_output()
plot_instance_pairs()
plot_instance_pairs_2()
plot_inverse_point()
plot_loss_trajectory()
swirl()
expand_axes()
flatten_along_axes()
move_axis()
swop_axes()