New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

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

Open

tuyunbin opened this issue Sep 19, 2020 · 0 comments

tuyunbin commented Sep 19, 2020

CCN/train/train.py

Line 242 in 2254049

    
           object_graph = masked_softmax(object_graph, diag_zero[None,...].cuda(), dim=1) + diag[None,...].cuda()

你好，按照论文里写的这里softmax是对每一行进行操作，obj_graph的维度是三维的，我的理解应该是对最后一维进行softmax，dim的设置应该是2或者-1吧？

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment