import torch
def search_index(key: torch.tensor, query: torch.tensor)
'''search indices in key of elements in query.'''
indices = torch.nonzero(torch.isin(key, query))
return indices
08-30
3861
09-15
9294
07-20
“相关推荐”对你有帮助么?
-
非常没帮助
-
没帮助
-
一般
-
有帮助
-
非常有帮助
提交