- 博客(2)
- 收藏
- 关注
RSS订阅原创 对torch.gather的简单理解
torch.gather(input,dim,index) input为输入矩阵，dim为维数，index为索引矩阵。这里根据维数和索引值去更换我需要的值，例如我dim=0，我就把这里的00，01，02的第一个0换成index输入的值。索引什么值呢，就是input的矩阵中对应的值，对应到x就是1,5,6，即gather的输出。torch.gather的输出size和index是一致的，这里是1×3一行三列的一个矩阵。看了一堆花里胡哨的，还是通过官方的定义看的明白。同理可扩展至更高维度。
2023-07-18 21:42:30
48
1
原创 powershell里conda init powershell - no action taken同时conda activate没有作用
用户名包含中文会出现什么问题我不知道 因为我没有 可以点原文链接查看。查看是否有powershell配置文件。
2023-07-07 09:56:03
937
2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人