def ValuesCount

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

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape

436def ValuesCount(+shape: Shape.Shape, +values: List<&2, F32>, result: Result<&2, &1, Error, Tensor<shape>>) -> Type:
437  match result:
438    case Fail{error}:
439      Unit
440    case Done{tensor}:
441      {Shape.logical_numel(Shape.dimensions(shape)) == List.length(&2, F32, values) : Nat}