MLU3系列的版本也还没有支持torch.cosine_similarity这个算子,您的需求我们会评估是否加入到后续的版本计划中,谢谢。
谢谢回复,后面有没有更新的版本适配呢,MLU3系列版本可以用吗?
虽然CNML有cnmlCreateCosSimilarityOp,但是寒武纪维护的PyTorch并没有适配torch.cosine_similarity这个算子,所以这里会出现如上的问题。
请登录后评论