मुझे निम्नलिखित त्रुटि मिलती है:

रनटाइम त्रुटि: स्केलर प्रकार की अपेक्षित वस्तु लंबी लेकिन स्केलर प्रकार फ्लोट के लिए तर्क # 3 'mat2' _th_addmm_out पर कॉल में मिला

मैं इस प्रकार टॉर्च.इनसम का उपयोग करता हूं:

mu = torch.einsum('ijl, akij -> akl', idxs, activation_map)

मुझे यह समझ में नहीं आता, क्योंकि दस्तावेज़ में वे फ्लोट टेंसर का भी उपयोग कर रहे हैं ( https://pytorch.org/docs/stable/generated/torch.einsum.html)। इसके अलावा एक लंबा टेंसर चुनना कोई विकल्प नहीं है, क्योंकि activation_map में सभी मान 0 और 1 के बीच हैं।

1
spadel 22 सितंबर 2020, 13:49

1 उत्तर

सबसे बढ़िया उत्तर

ऐसा लगता है कि आपका पहला तर्क idxs Long प्रकार का है।
torch.einsum के सभी इनपुट टेंसर Float होने चाहिए।

1
Shai 22 सितंबर 2020, 13:52