def Zeros

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

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/elementwise_totality.bend as Totality

89def Zeros(+shape: Shape.Shape, result: Result<&2, &1, Core.Error, Core.Dynamic>) -> Type:
90  match result:
91    case Fail{error}:
92      Empty
93    case Done{gradient}:
94      {Totality.dimensions(gradient) == Shape.dimensions(shape) : List<&2, U32>} &
95        (@+index: U32 -> @inside: Inside(shape, index) ->
96          {Core.checked_observe_dynamic(gradient, index) == Some{0.0} : Maybe<&2, F32>})