def output_dimensions

bendtorch-core@0.1.0.0 · src/autograd/reduce_fold_LAWS.bend:42

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/reduce.bend as Reduce

42def output_dimensions(result: Core.Dynamic & Result<&2, &1, Reduce.Error, Core.Dynamic>) -> Maybe<&2, List<&2, U32>>:
43  (input, outcome) = result
44  match outcome:
45    case Fail{error}:
46      None{}
47    case Done{Core.Packed{shape, tensor}}:
48      Some{Shape.dimensions(shape)}