Shimingyi
commented
3 years ago a is just a temporary variant for debugging, I will delete it in next commit.
The operation is running on the batchwise, and when testing, the batchsize is 1. If you output the shape of alpha here, you will find the fist dimension value is 1. so we can use [0] to get it.
First, a is defined ,but not use . Second, alphas is a list , why only use alphas[0] ??
I look forward to your answer !!!! Thank you !!!