torch_mist.distributions.transforms.implementations.linear

Module Contents

Classes

ConditionedLinear
Linear
ConditionalLinear

class torch_mist.distributions.transforms.implementations.linear.ConditionedLinear(params, epsilon=1e-06)

Bases: torch.distributions.Transform

domain

codomain

bijective = True

_call(x)

_inverse(y)

log_abs_det_jacobian(x, y)

class torch_mist.distributions.transforms.implementations.linear.Linear(input_dim, loc=None, scale=None, initial_scale=None, epsilon=1e-06)

Bases: ConditionedLinear, pyro.distributions.TransformModule

_params()

class torch_mist.distributions.transforms.implementations.linear.ConditionalLinear(net, loc=None, scale=None, initial_scale=None, epsilon=1e-06, skip_connection=False)

Bases: pyro.distributions.ConditionalTransformModule

domain

codomain

bijective = True

_params(context)

condition(context)