MCPcopy Create free account
hub / github.com/tensorflow/lingvo / _AddAttenProbsSummary

Method _AddAttenProbsSummary

lingvo/tasks/mt/decoder.py:315–329  ·  view source on GitHub ↗

Add summary of attention probs. Args: source_paddings: source padding, of shape [src_len, src_batch]. targets: A dict of string to tensors representing the targets one try to predict. Each tensor in targets is of shape [tgt_batch, tgt_len]. atten_probs: a list of atten

(self, source_paddings, targets, atten_probs)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 3

ComputePredictionsMethod · 0.45
_AddAttenProbsSummaryMethod · 0.45
_FPropMethod · 0.45

Tested by

no test coverage detected