law sum_dims_kept_start

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

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

63# The kept dimensions count axes from 0, and set each reduced axis to 1.
64law sum_dims_kept_start:
65  for +axes: List<&2, U32>
66  for +dimensions: List<&2, U32>
67  {SumDims.kept_dimensions(axes, dimensions) == SumDims.kept_from(axes, dimensions, 0) : List<&2, U32>}