If I want to customize a 2048 length vector as a parameter in the model, how do I define it in model’s __init__ function?Its operation in forward() is:x = F.relu(x * self.W_sub1).unsqueeze(1).
I have used self.W_sub1 = nn.Parameter(torch.randn(2048)) to define it,but got:
`TypeError: mul received an invalid combination of arguments - got (Parameter), but expected one of:

(float value)
didn’t match because some of the arguments have invalid types: (Parameter)`