Zeroes out the first time step. Args: token_embs: [batch, time, model_dim] embeding lookups batch: Batch size scalar target_time: Target sequence length scalar. Returns: modified token_embs with the first time step zeroed out.
(self, token_embs, batch, target_time)
source not stored for this graph (policy: none)