torch_mist.distributions.transforms.implementations.permute

Module Contents

Classes

Permute

class torch_mist.distributions.transforms.implementations.permute.Permute(permutation, *, dim=-1, cache_size=1)

Bases: pyro.distributions.transforms.Permute, pyro.distributions.TransformModule

update_device()

_call(x)

_inverse(y)

log_abs_det_jacobian(x, y)