MCPcopy Create free account
hub / github.com/XuezheMax/flowseq / log_probability

Method log_probability

flownmt/modules/priors/prior.py:161–195  ·  view source on GitHub ↗

Args: z: Tensor tensor of latent code [batch, length, nz] tgt_mask: Tensor tensor of target masks [batch, length] src: Tensor source encoding [batch, src_length, hidden_size] ctx: Tensor

(self, z: torch.Tensor, tgt_mask: torch.Tensor,
                        src: torch.Tensor, ctx: torch.Tensor, src_mask: torch.Tensor,
                        length_loss: bool = True)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 2

translate_iwMethod · 0.45
translate_lossMethod · 0.45

Calls 2

length_lossMethod · 0.95
bwdpassMethod · 0.80

Tested by

no test coverage detected