law sum_dims_refused

bendtorch-core@0.1.0.0 · src/autograd/sum_dims_LAWS.bend:145

import Base
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
import bendtorch-core@0.1.0.0/src/autograd/sum_dims.bend as SumDims

145# A refused plan fails the sum with its error and hands the input back.
146law sum_dims_refused:
147  for input: Core.Dynamic
148  for +axes: List<&2, U32>
149  for keepdim: Bool
150  for error: SumDims.Error
151  for refused: {SumDims.plan(Totality.dynamic_shape(input), axes) == Fail{error} : Result<&2, &2, SumDims.Error, SumDims.Plan>}
152  {SumDims.sum_dims(input, axes, keepdim) == (input, Fail{error}) : Core.Dynamic & Result<&2, &1, SumDims.Error, Core.Dynamic>}