If anyone came across this post via a google search, I suggest they check out the following github issue:

  
      github.com/pytorch/pytorch
  
  
    



Issue: Support Batch Dot Product 337


	
	opened by sidazhang
	on 2019-03-14
	
	
	


🚀 Feature
Support batch dot product
Motivation
Commonly used operation
Alternatives
Currently, we can do this with bmm. But I think it makes sense to just...


 	feature
 	triaged


  
  
    
    
  
  


Given two batches of vectors A,B, it is the fastest to just compute (A*B).sum(-1)