快捷键

torch.nn.functional.bilinear

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

对传入数据进行双线性变换: y=x1TAx2+by = x_1^T A x_2 + b

形状:

  • input1: (N,,Hin1)(N, *, H_{in1}) 其中 Hin1=in1_featuresH_{in1}=\text{in1\_features}* 表示任意数量的附加维度。除了输入的最后一个维度外,所有维度都应该相同。

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

  • 重量: (out_features,in1_features,in2_features)(\text{out\_features}, \text{in1\_features}, \text{in2\_features})

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

  • 输出: (N,,Hout)(N, *, H_{out}) 其中 Hout=out_featuresH_{out}=\text{out\_features} 以及除了最后一个维度之外的所有维度与输入形状相同。


© 版权所有 PyTorch 贡献者。

使用 Sphinx 构建,并使用 Read the Docs 提供的主题。

文档

PyTorch 的全面开发者文档

查看文档

教程

深入了解初学者和高级开发者的教程

查看教程

资源

查找开发资源并获得您的疑问解答

查看资源