chickadee » nanograd » squeeze

squeeze tensor #!key (axis #f)procedure

Removes dimensions of size 1. With axis, removes only that dimension, which must have size 1; negative values count from the end. If every dimension is removed, the result has shape (1).