chickadee » nanograd » make-am-flatten

make-am-flatten #!key (name Flatten)procedure

Flattens each sample: (N, ...) -> (N, product of the other dimensions). 1-D and 2-D inputs are passed through unchanged.