Make accessing W elements cost less in kl-div loss evaluation.
Make accessing W elements cost less in kl-div loss evaluation.

fengjingchao/taskgraph,plutoshe/taskgraph,taskgraph/taskgraph,BaiGang/taskgraph,BaiGang/taskgraph,kod3r/taskgraph,dzch/taskgraph,taskgraph/taskgraph,kod3r/taskgraph,5kg/taskgraph,plutoshe/taskgraph,5kg/taskgraph,fengjingchao/taskgraph,dzch/taskgraph