torch.nn.functional.bilinear
-
torch.nn.functional.bilinear(input1, input2, weight, bias=None) → Tensor
对传入数据进行双线性变换: y=x1TAx2+b
形状:
input1: (N,∗,Hin1) 其中 Hin1=in1_features 和 ∗ 表示任意数量的附加维度。除了输入的最后一个维度外,所有维度都应该相同。
输入 2: (N,∗,Hin2) 其中 Hin2=in2_features
重量: (out_features,in1_features,in2_features)
偏置: (out_features)
输出: (N,∗,Hout) 其中 Hout=out_features 以及除了最后一个维度之外的所有维度与输入形状相同。