Revise:I want y=(A*B) * x where x is a vector as one input and A is a matrix as another input and B is the trainable elementwise product matrix I desire. So far, I did :
self.net = nn.Sequential(A, )
And I am confused how to step forward.