ML_modules.get_flow#

Back to Module page

Summary#

Instantiate a simple Masked Autoregressive normalizing flow.

Signature#

def get_flow(d_in = 3, d_hidden = 64, d_context = 2, n_layers = 4)
Input variables#

Name

Type

Default

Description

d_in

3

d_hidden

64

d_context

2

n_layers

4

Output variables#

Return annotation

Docstring type

Description

None

Docstring#

Instantiate a simple Masked Autoregressive normalizing flow.