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

Method predict_length

flownmt/modules/priors/prior.py:28–41  ·  view source on GitHub ↗

Args: ctx: Tensor tensor [batch, features] src_mask: Tensor tensor for source mask [batch, src_length] topk: int (default 1) return top k length candidates for each src sentence Returns: LongTensor1,

(self, ctx: torch.Tensor, src_mask: torch.Tensor, topk: int = 1)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 2

sampleMethod · 0.95
reconstructMethod · 0.80

Calls 1

predictMethod · 0.45

Tested by

no test coverage detected