Fix bug in linspace model generation (#72027)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/72027
Test Plan: build is successful
Reviewed By: cccclai
Differential Revision: D33858104
fbshipit-source-id: 77a5d4ab15efc21f128efbba1fcce63c6dea8018
(cherry picked from commit b03b9d6b2f5435afd22ac66155dd8542e55bb5da)
