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)