Can you tell me if these two pytorch operations are equivalent: "torch.bmm(attention_slice, value)" and "einsum('b i j, b j d -> b i d', attention_slice, value)"