def natural_step
bendtorch-core@0.1.0.0 · src/tensor/elementwise.bend:217
import bendtorch-core@0.1.0.0/src/tensor/elementwise.bend as Elementwise
import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
217def natural_step(+operation: Operation, +plan: Shape.BroadcastPlan, +cursor: Nat, state: State<Shape.output_shape(plan)>) ->
218 NaturalStep<Shape.output_shape(plan), cursor>:
219 +index = U32.from_nat(cursor)
220 natural_step_decided(operation, plan, cursor, index, state, Shape.bool_case(Nat.is_eq(U32.to_nat(index), cursor)))