chop
chop copied to clipboard
Constraints should take dim argument
... to specify which dimensions are batch dims, and which are data dims.