Hi,
You can use “```” to format your code properly here, that makes it easier to read.
Also, you should use the latest version of pytorch and so Variables are not needed anymore: they have been merged with Tensors.
Your tensor treino_x is of size 10x3 so represent a batch size of 10 and feature size of 3.
Your Linear(10, 1) layer will convert a feature size of 10 to a feature size of 1.
So there is a size mismatch where a linear layer expects a tensor with feature size 10 and you give it one of size 3.