def squeezed

bendtorch-core@0.1.0.0 · src/autograd/sum_dims.bend:122

import Base
import bendtorch-core@0.1.0.0/src/tensor/core.bend as Core
import bendtorch-core@0.1.0.0/src/autograd/reduce.bend as Reduce

122def squeezed(+dropped: List<&2, U32>, keepdim: Bool, result: Result<&2, &1, Reduce.Error, Core.Dynamic>) ->
123  Result<&2, &1, Error, Core.Dynamic>:
124  match result:
125    case Fail{error}:
126      Fail{ReduceError{error}}
127    case Done{output}:
128      shaped(dropped, keepdim, output)