动手学强化学习(八.1):torch.gather

tensor.gather()的作用就是按照索引取对应的数据出来。之前看图解PyTorch中的torch.gather函数,那个图示看得a56爆大奖在线娱乐有点懵逼,a56爆大奖在线娱乐自己画了两张图总结了一下规律来理解一下。 首先新建一个3*3的二维矩阵。 import torch ? t1 = torch.tensor([[1,
posted @ 2024-03-05 17:01  jasonzhangxianrong  阅读(24)  评论(0编辑  收藏  举报