[ 0.01601785  0.14540915  0.08909235  0.0281623  -0.01601157 -0.01005557]
tensor([-0.5321,  0.6286, -0.4779, -0.1251, -0.5960,  0.0382], device='cuda:0',
       grad_fn=<MulBackward0>)
