BilinearTensorProduct
该接口用于构建 BilinearTensorProduct
类的一个可调用对象,具体用法参照 。双线性乘积计算式子如下。
式中,
-
: 第一个输入,分别包含M个元素,维度为
-
:第二个输入,分别包含N个元素,维度为
WiWi :第i个学习到的权重,维度为 [M,N][M,N]
yTyT : yy 的转置
参数:
input1_dim (int) – 第一个输入的维度大小。
input1_dim (int) – 第二个输入的维度大小。
output_dim (int) – 输出的维度。
param_attr (ParamAttr) – 指定权重参数属性的对象。默认值为None,表示使用默认的权重参数属性。具体用法请参见 。
bias_attr (ParamAttr) – 指定偏置参数属性的对象。默认值为None,表示使用默认的偏置参数属性。具体用法请参见 ParamAttr。
dtype (str, 可选) - 数据类型,可以为”float32”或”float64”。默认值为”float32”。
返回:Tensor, 维度为[batch_size, size]的2D Tensor,数据类型与输入数据类型相同。
代码示例
weight
本层的可学习参数,类型为
bias