law sum_dims_dropped_kept

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

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

91law sum_dims_dropped_kept:
92  for +axes: List<&2, U32>
93  for +extent: U32
94  for +rest: List<&2, U32>
95  for +axis: U32
96  for kept: {SumDims.reduces(axes, axis) == False{} : Bool}
97  {SumDims.dropped_from(axes, extent <> rest, axis) == extent <> SumDims.dropped_from(axes, rest, U32.add(axis, 1)) : List<&2, U32>}