def planned_source

bendtorch-core@0.1.0.0 · src/autograd/broadcast_LAWS.bend:27

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
import bendtorch-core@0.1.0.0/src/tensor/projection_LAWS.bend as Projection

27def planned_source(input: Core.Dynamic, +index: U32, plan: Result<&2, &2, Shape.Error, Shape.BroadcastPlan>) -> Maybe<&2, F32>:
28  match plan:
29    case Fail{error}:
30      None{}
31    case Done{+plan}:
32      source_value(input, Projection.offset_spec(plan, U32.to_nat(index)))