API symbolsnn
Copy
tensorplay.nn.functional.linear_cross_entropy
- tensorplay.nn.functional.linear_cross_entropy(input: TensorBase, linear_weight: TensorBase, target: TensorBase, *, linear_bias=None, weight=None, reduction: str = 'mean', ignore_index=None, label_smoothing: float = 0.0, options=None) TensorBase[source]
Compute cross entropy between
input, transformed linearly, and target.Equivalent to
cross_entropy(linear(input, linear_weight), target, **kwargs)(reference path; chunked/fused options are ignored).
Help improve this page
Found an error, an unclear step, or a missing example?
Was this page helpful?
