def output_dimensions

bendtorch-core@0.1.0.0 · src/tensor/transform_observation.bend:88

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

88def output_dimensions(result: Result<&2, &1, Transforms.Error, Core.Dynamic>) -> Maybe<&2, List<&2, U32>>:
89  match result:
90    case Fail{error}:
91      None{}
92    case Done{output}:
93      Some{Transforms.dynamic_dimensions(output)}