Hi all,
I have two tensors with shape [B1, N] and [B2, N], N is the vector length.
I want to get a similarity metric with shape [B1, B2] with the current API.
Can I do that at once? I found that it always rises shape mismatching error.

image.png962×529 61.6 KB