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

Method _ZeroOutFirstTimeStep

lingvo/tasks/mt/decoder.py:657–677  ·  view source on GitHub ↗

Zeroes out the first time step. Args: token_embs: [time, batch, model_dim] embeding lookups. batch: Batch size scalar. time: Target sequence length scalar. Returns: modified token_embs with the first time step zeroed out.

(self, token_embs, batch, time)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

ComputePredictionsMethod · 0.95

Calls

no outgoing calls

Tested by

no test coverage detected