PaddlePaddle

 paddle.static / nn / bilinear_tensor_product


bilinear_tensor_product

paddle.static.nn. bilinear_tensor_product ( x, y, size, act=None, name=None, param_attr=None, bias_attr=None ) [源代码]

该层对两个输入执行双线性张量积。

例如:

\[out_{i} = x * W_{i} * {y^\mathrm{T}}, i=0,1,...,size-1\]
在这个公式中:
  • \(x\): 第一个输入,包含 \(M\) 个元素,形状为 [batch_size, M]。

  • \(y\): 第二个输入,包含 \(N\) 个元素,形状为 [batch_size, N]。

  • \(W_{i}\): 第 \(i\) 个被学习的权重,形状是 [M, N]。

  • \(out_{i}\): 输出的第 \(i\) 个元素,形状是 [batch_size, size]。

  • \(y^\mathrm{T}\): \(y_{2}\) 的转置。

参数:
  • x (Variable): 2-D 输入张量,形状为 [batch_size, M], 数据类型为 float32 或 float64。

  • y (Variable): 2-D 输入张量,形状为 [batch_size, N],数据类型与 x 一致。

  • size (int): 此层的维度。

  • act (str, 可选): 应用到该层输出的激活函数。

  • name (str,可选) – 具体用法请参见 cn_api_guide_Name ,一般无需设置,默认值为 None。

  • param_attr (ParamAttr,可选) :指定权重参数属性的对象。默认值为 None,表示使用默认的权重参数属性。具体用法请参见 ParamAttr

  • bias_attr (ParamAttr,可选) : 指定偏置参数属性的对象。默认值为 None,表示使用默认的偏置参数属性。具体用法请参见 ParamAttr

返回: Variable, 一个形为 [batch_size, size] 的 2-D 张量。

代码示例:

import paddle
paddle.enable_static()
layer1 = paddle.static.data("t1", shape=[-1, 5], dtype="float32")
layer2 = paddle.static.data("t2", shape=[-1, 4], dtype="float32")
tensor = paddle.static.nn.bilinear_tensor_product(x=layer1, y=layer2, size=1000)

此页内容是否对您有帮助