CCN icon indicating copy to clipboard operation
CCN copied to clipboard

Published 20 hours ago • verified publisher icon AmingWu

请问softmax的维度这里为什么不是2或者-1

Open tuyunbin opened this issue 4 years ago • 0 comments

https://github.com/AmingWu/CCN/blob/2254049e5b090952406f14fbf4a63b4e30e99a17/train/train.py#L242 你好，按照论文里写的这里softmax是对每一行进行操作，obj_graph的维度是三维的，我的理解应该是对最后一维进行softmax，dim的设置应该是2或者-1吧？

tuyunbin avatar Sep 19 '20 03:09 tuyunbin