<?xml version="1.0" encoding="utf-8" ?><rss version="2.0"><channel><title><![CDATA[AI_Geeker的博客]]></title><description><![CDATA[]]></description><link>https://blog.youkuaiyun.com/AI_Geeker</link><language>zh-cn</language><generator>https://blog.youkuaiyun.com/</generator><copyright><![CDATA[Copyright &copy; AI_Geeker]]></copyright><item><title><![CDATA[torch.gather函数的说明]]></title><link>https://blog.youkuaiyun.com/AI_Geeker/article/details/112359415</link><guid>https://blog.youkuaiyun.com/AI_Geeker/article/details/112359415</guid><author>AI_Geeker</author><pubDate>Fri, 08 Jan 2021 14:09:37 +0800</pubDate><description><![CDATA[torch.gather函数的理解
一.函数说明
torch.gather(input, dim, index, *, sparse_grad=False, out=None) → Tensor
dim=0 时，output[i][j][k] = input[a][j][k] ，
dim=1时，output[i][j][k] = input[i][a][k]
dim=2时，output[i][j][k] = input[i][j][a]
上述input索引中的a=index[i][j][k]。
说明：out]]></description><category></category></item></channel></rss>