实现nn.Bi线性层

Question

任何人都可以帮助我理解

nn.Bilinear

的实施吗 根据文档，此函数实现 y = x₁^T * A * x₂ 取

x1 = (100,20)

，

x2 = (100,30')

，假设

output_features = 50

。矩阵

A

的维度为

[50,20,30]

。 我发现很难将这些矩阵相乘以获得

output = [100,50]

根据 x₁、x₂ 和

A

矩阵的大小，根据 y = x₁^T * A * x₂ ，乘法似乎不兼容。我在这里缺少什么？

Answer 1

 T

y =x1Ax2+b。 PyTorch 就是这样