def held_dimensions

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

import Base
import bendtorch-core@0.1.0.0/src/tensor/elementwise_totality.bend as Totality
import bendtorch-core@0.1.0.0/src/autograd/gradient.bend as Gradient

27def held_dimensions(slot: Gradient.Slot) -> Maybe<&2, List<&2, U32>>:
28  match slot:
29    case Gradient.Unset{}:
30      None{}
31    case Gradient.Held{gradient}:
32      Some{Totality.dimensions(gradient)}