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})$
• output: $(N, *, H_{out})$ где $H_{out}=\text{out\_features}$ и все измерения, кроме последнего, имеют такую же форму, как у входа.