def after_count_pair
import bendtorch-core@0.1.0.0/src/tensor/observation.bend as Observationimport 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
75def after_count_pair(+shape: Shape.Shape,
76 pair: Core.Tensor<shape> & Result<&2, &2, Core.Error, F32>) -> Maybe<&2, U32>:
77 (tensor, result) = pair
78 count_pair(Core.count(Core.pack(shape, tensor)))