The output shape of [15, 1] is a bit weird, since it should be [batch_size, 17*batch_size] based on your model definition.
You can define the output shape via the out_features of the linear layer.
That being said, it’s also unusual to define a specific shape relative to the batch size, as the model definition is independent of the batch size in a standard scenario.