def output_count

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

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
import bendtorch-core@0.1.0.0/src/tensor/observation.bend as Observation

120def output_count(result: Result<&2, &1, Transforms.Error, Core.Dynamic>) -> Maybe<&2, U32>:
121  match result:
122    case Fail{error}:
123      None{}
124    case Done{output}:
125      Observation.count_pair(Core.count(output))