csarofeen/pytorch

squeeze prim should accept a sequence of dimensions to squeeze, not just one

Closed this issue · 0 comments

mruberry commented

Ex. prims.squeeze(a, a.shape, (0, 1))