torch.nn.functional.bilinear

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

對輸入資料應用雙線性轉換： $y = x_1^T A x_2 + b$

形狀 (Shape)

• 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}$ 且除了最後一個維度之外，所有維度的形狀都與輸入相同。