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

Method _FPropSoftmax

lingvo/tasks/mt/decoder.py:223–255  ·  view source on GitHub ↗

Computes cross-entropy loss given the softmax input, labels and weights. Args: theta: A `.NestedMap` object containing weights' values of this layer and its children layers. softmax_input: A tensor of shape [time, batch, p.softmax.input_dim]. target_labels: A matrix of

(self,
                    theta,
                    softmax_input,
                    target_labels,
                    target_weights,
                    target_paddings,
                    target_segment_ids=None,
                    time_axis=0)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 2

ComputeLossMethod · 0.95
ComputeLossMethod · 0.45

Calls 2

_ComputeXentLossMethod · 0.95

Tested by

no test coverage detected