torch:squeeze
(torch:squeeze a &key axis)
Differentiable extent-1 axis removal (linalg:squeeze): all of them with no :axis, else only the named axis (or list of axes). Squeezing every axis away yields the scalar tensor.
(torch:squeeze a &key axis)
Differentiable extent-1 axis removal (linalg:squeeze): all of them with no :axis, else only the named axis (or list of axes). Squeezing every axis away yields the scalar tensor.