Suppose I had tensors X and Y which are both (batch_size, d) dimensional. I would like to find the (batch_size x 1) tenso
X
Y
(batch_size, d)
(batch_size x 1)