关于loss function的疑问
lizhihao6 opened this issue · comments
lizhihao6 commented
Morvan commented
谢谢你指出问题,是的,td
的确有维度问题,td
的维度需要和a
一样, 我根据这个修改了代码。
lizhihao6 commented
看到您修改的代码是
exp_v = m.log_prob(a) * td.detach().squeeze()
想请教一下,使用squeeze和[0]有什么区别吗?
exp_v = m.log_prob(a) * td.detach()[0]
Morvan commented
其实在这个例子中是没差别的。squeeze主要是将所有为1的维度去掉。比如(5,1,2) 变成(5,2)
lizhihao6 commented
好的,谢谢莫烦:)
发自我的 iPad
在 2018年8月20日,下午6:40,Morvan <notifications@github.com<mailto:notifications@github.com>> 写道:
其实在这个例子中是没差别的。squeeze主要是将所有为1的维度去掉。比如(5,1,2) 变成(5,2)
—
You are receiving this because you authored the thread.
Reply to this email directly, view it on GitHub<#5 (comment)>, or mute the thread<https://github.com/notifications/unsubscribe-auth/AVJFRKejLGOYkmQLft61uFtYOL7Njd1Kks5uSpIHgaJpZM4WDFIy>.