torch.bilinear

torch.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}$, а все измерения, кроме последнего, имеют ту же форму, что и входные данные.