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

Method _AddAttenProbsImageSummary

lingvo/tasks/mt/decoder.py:343–379  ·  view source on GitHub ↗

Add image 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

(self, source_paddings, targets, atten_probs)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

_AddAttenProbsSummaryMethod · 0.95

Calls 1

AddSubplotMethod · 0.80

Tested by

no test coverage detected