Fixed the error in conv1d example (#57356)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/51225
Pull Request resolved: https://github.com/pytorch/pytorch/pull/57356
Reviewed By: albanD
Differential Revision: D28173174
Pulled By: malfet
fbshipit-source-id: 5e813306f2e2f7e0412ffaa5d147441134739e00