查询numpy中的最大值索引

print(np.where(test_user_embedding == np.max(test_user_embedding))) 

你可能感兴趣的:(python,python)