Pytorch gather. 按照同样的思路,试着使用torch.gather(): import torch.nn as nn import torch t=np.random.randint(1,10,[4,5]) index ...
確定! 回上一頁