def numel
import bendtorch-core@0.1.0.0/src/shape/core.bend as Coreimport Base
130def numel(+dims: List<&2, Nat>) -> Nat:
131 match dims:
132 case Nil{}:
133 1n
134 case head <> tail:
135 Nat.mul(head, numel(tail))
import bendtorch-core@0.1.0.0/src/shape/core.bend as Coreimport Base
130def numel(+dims: List<&2, Nat>) -> Nat:
131 match dims:
132 case Nil{}:
133 1n
134 case head <> tail:
135 Nat.mul(head, numel(tail))