I have two 3D tensor and I want to use one's top k indices get another top k. ... pytorch's topk function will give me the following.
確定! 回上一頁