Notes_200124_074441_0f5_11000×1542 59.6 KB

Thanks a lot for your reply. It does partially answer my question. I have attached an image of what I am intending to achieve.
There are 5 document tensors (each is 3 x 10).

I want to apply weights to individual rows
Sum these weighted rows to get a document vector which then goes as input to the neural network.
At the same time, I want the initial weights w to be updated too.

Can you please help me figure this out, thanks?