torch.nn.functional.bilinear

torch.nn.functional.bilinear(input1, input2, weight, bias=None) → Tensor

应用双线性变换到输入数据：$y = x_1^T A x_2 + b$

形状

• input1：$(N, *, H_{in1})$，其中$H_{in1}=\text{in1\_features}$，而$*$ 表示任意数量的额外维度。除了最后一个维度外，输入的维度应相同。

• input2：$(N, *, H_{in2})$，其中$H_{in2}=\text{in2\_features}$

• weight：$(\text{out\_features}, \text{in1\_features}, \text{in2\_features})$

• bias：$(\text{out\_features})$

• 输出： $(N, *, H_{out})$ 其中 $H_{out}=\text{out\_features}$ 并且除最后一个维度外，所有维度的形状都与输入相同。