fix cosine similarity dimensionality check#66191
Conversation
CI Flow Status⚛️ CI FlowRuleset - Version:
You can add a comment to the PR and tag @pytorchbot with the following commands: # ciflow rerun, "ciflow/default" will always be added automatically
@pytorchbot ciflow rerun
# ciflow rerun with additional labels "-l <ciflow/label_name>", which is equivalent to adding these labels manually and trigger the rerun
@pytorchbot ciflow rerun -l ciflow/scheduled -l ciflow/slowFor more information, please take a look at the CI Flow Wiki. |
🔗 Helpful links
💊 CI failures summary and remediationsAs of commit 9c8dcb4 (more details on the Dr. CI page): 💚 💚 Looks good so far! There are no failures yet. 💚 💚 This comment was automatically generated by Dr. CI (expand for details).Follow this link to opt-out of these comments for your Pull Requests.Please report bugs/suggestions to the (internal) Dr. CI Users group. |
| and broadcastable with x1 at other dimensions). | ||
| dim (int, optional): Dimension of vectors. Default: 1 | ||
| x2 (Tensor): Second input. | ||
| dim (int, optional): Dimension along which cosine similarity is computed. Default: 1 |
There was a problem hiding this comment.
aren't the docs wrong below?
Inputisn't a thing, there are two inputs.Dis now based on the output shape, not the "Input"- The notation doesn't make much sense, are
\ast_1\and `ast_2` not supposed to be the same thing? They can't be if D is a value...
There was a problem hiding this comment.
Yeah, dimensions below don't make any sense.
|
@ngimel has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator. |
Fixes #66086