law sum_dims_kept_step

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

import Base
import bendtorch-core@0.1.0.0/src/autograd/sum_dims.bend as SumDims

69law sum_dims_kept_step:
70  for +axes: List<&2, U32>
71  for +extent: U32
72  for +rest: List<&2, U32>
73  for +axis: U32
74  {SumDims.kept_from(axes, extent <> rest, axis) ==
75    Bool.pick(U32, SumDims.reduces(axes, axis), 1, extent) <> SumDims.kept_from(axes, rest, U32.add(axis, 1)) : List<&2, U32>}