def entropy_shaped

bendtorch-core@0.1.0.0 · src/autograd/tensor_backward.bend:1645

import bendtorch-core@0.1.0.0/src/autograd/tensor_backward.bend as TensorBackward
import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/core.bend as Core
import bendtorch-core@0.1.0.0/src/autograd/loss.bend as Loss

1645def entropy_shaped(+a: Nat, +targets: List<&2, U32>, entries: List<&1, Entry>, pair: Core.Dynamic & Shape.Shape) ->
1646  Tape & Result<&2, &2, Error, Nat>:
1647  (input, +shape) = pair
1648  loss_kept(CrossEntropy{a, shape}, entries, Loss.cross_entropy(input, targets))

Uses (14)

Used by (1)