def Fits

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

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
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

75def Fits(+shape: Shape.Shape, slot: Gradient.Slot) -> Type:
76  match slot:
77    case Gradient.Unset{}:
78      Unit
79    case Gradient.Held{gradient}:
80      {Totality.dimensions(gradient) == Shape.dimensions(shape) : List<&2, U32>}