def count_outcome

bendtorch-core@0.1.0.0 · src/tensor/core_PROOF.bend:53

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

53def count_outcome(+shape: Shape.Shape, +values: List<&2, F32>,
54  equal: {Shape.logical_numel(Shape.dimensions(shape)) == List.length(&2, F32, values) : Nat},
55  result: Result<&2, &1, Core.Error, Core.Tensor<shape>>) -> Core.ValuesCount(shape, values, result):
56  match result:
57    case Fail{error}:
58      Unit{}
59    case Done{tensor}:
60      equal