ฉันมีพารามิเตอร์เทนเซอร์คบเพลิง 4 มิติที่กำหนดดังนี้:
nn.parameter.Parameter(data=torch.Tensor((13,13,13,13)), requires_grad=True)
และเทนเซอร์สี่ตัวพร้อมสลัว (batch_size,13) (หรือเทนเซอร์หนึ่งตัวพร้อมสลัว (batch_size,4,13)) อยากได้เทนเซอร์ที่มี dims (batch_size) เท่ากับสูตรท้ายภาพนี้ : [แก้ไข: รูปแรกผมทำผิด แก้ไขให้แล้ว] 
ฉันเคยเห็นในเอกสารประกอบของ torch เกี่ยวกับฟังก์ชัน tensordot แต่ฉันไม่สามารถทำให้มันทำงานได้ด้วยตัวเอง
เมื่อใดก็ตามที่คุณมีผลิตภัณฑ์เทนเซอร์ตลกๆ torch.einsum (หรือ numpy.einsum) คือเพื่อนของคุณ:
batch_size = 5
A = torch.rand(13, 13, 13, 13)
a = torch.rand(batch_size, 13)
b = torch.rand(batch_size, 13)
c = torch.rand(batch_size, 13)
d = torch.rand(batch_size, 13)
B = torch.einsum('ijkl,bi,bj,bk,bl->b', A, a, b, c, d)
torch.einsum เป็นส่วนหนึ่งของการดำเนินการพื้นฐานของ pytorch ไม่เพียงใช้งานได้กับ cpu/gpu เท่านั้น แต่ยังเผยแพร่การไล่ระดับสีไปยังเทนเซอร์อินพุตอีกด้วย
- person Shai; 20.01.2021
iแต่มันแตกต่างจากตัวแปรปิดเสียงที่คุณดำเนินการผลรวมนอกสุดใช่ไหม - person trialNerror schedule 20.01.2021result = (A[None, :, :, :, :] * X[:, :, None, None, None] * Y[:, None, :, None, None] * Z[:, None, None, :, None] * T[:, None, None, None, :]).flatten(1).sum(dim=1)เป็นสิ่งที่คุณต้องการใช่ไหม? ถ้าเป็นเช่นนั้นฉันสามารถโพสต์คำตอบพร้อมคำอธิบายได้ - person jodag schedule 20.01.2021Noneใช้เพื่อแทรกมิติข้อมูลแบบรวมเช่นเดียวกับตัวเลข นี่เป็นการออกอากาศที่เทียบเท่ากับสำนวน einsum ของคำตอบของ Shai - person jodag schedule 20.01.2021