Simple Linear Regression - Pytorch Typing CST Test