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}$ 并且 $*$ 表示任意数量的额外维度。输入的所有维度，除了最后一个维度之外，都应该是相同的。

• 输入2: $(N, *, H_{in2})$ 其中 $H_{in2}=\text{in2\_features}$

• 权重: $(\text{out\_features}, \text{in1\_features}, \text{in2\_features})$

• 偏置: $(\text{out\_features})$

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