def dims_found

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

import Base
import bendtorch-core@0.1.0.0/src/tensor/core.bend as Core
import bendtorch-core@0.1.0.0/src/tensor/signed.bend as Signed

1404def dims_found(+a: Nat, +axes: List<&2, Signed.Position>, keepdim: Bool, pair: Found()) -> Tape & Result<&2, &2, Error, Nat>:
1405  (entries, input) = pair
1406  match input:
1407    case Fail{error}:
1408      (Tape{entries}, Fail{UnknownNode{}})
1409    case Done{input}:
1410      dims_shaped(a, axes, keepdim, entries, Core.shape(input))