Solved by cbats in post #3 
              
              
                Sorry, I made a mistake. Actually I solve it by specialized the Broadcast’s backward function(which is a reduce operation). 
Thanks.