def count

bendtorch-core@0.1.0.0 · src/tensor/observation.bend:22

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

22def count(+shape: Shape.Shape, result: Result<&2, &1, Core.Error, Core.Tensor<shape>>) -> Maybe<&2, U32>:
23  match result:
24    case Fail{error}:
25      None{}
26    case Done{tensor}:
27      count_pair(Core.count(Core.pack(shape, tensor)))