Remove AssertNDim, which doesn't seem to be used.
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/17978
Reviewed By: colesbury
Differential Revision: D14438845
Pulled By: gchanan
fbshipit-source-id: 106650c37fb1885201eaef27cb6d86b49ef27976