def shaped

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

import Base
import bendtorch-core@0.1.0.0/src/tensor/core.bend as Core
import bendtorch-core@0.1.0.0/src/tensor/transforms.bend as Transforms

115def shaped(+dropped: List<&2, U32>, keepdim: Bool, output: Core.Dynamic) -> Result<&2, &1, Error, Core.Dynamic>:
116  match keepdim:
117    case True{}:
118      Done{output}
119    case False{}:
120      transformed(Transforms.reshape_dynamic(output, dropped))