inner
Takes a tensor of shape T, a shape B, and a tensor of shape T and returns a tensor of shape T which is the inner product of the two.
Takes a tensor of shape T, a shape B, and a tensor of shape T and returns a tensor of shape T which is the inner product of the two.