提交 319618e9 编写于 作者: Q Qiao Longfei

optimize comment, add unit test test=develop

上级 53781fc0
...@@ -8063,13 +8063,13 @@ def bilinear_tensor_product(x, ...@@ -8063,13 +8063,13 @@ def bilinear_tensor_product(x,
For example: For example:
.. math:: .. math::
y_{i} = x * W_{i} * {y^\mathrm{T}}, i=0,1,...,K-1 out{i} = x * W_{i} * {y^\mathrm{T}}, i=0,1,...,size-1
In this formular: In this formular:
- :math:`x`: the first input contains M elements. - :math:`x`: the first input contains M elements, shape is [batch_size, M].
- :math:`y`: the second input contains N elements. - :math:`y`: the second input contains N elements, shape is [batch_size, N].
- :math:`y_{i}`: the i-th element of y.
- :math:`W_{i}`: the i-th learned weight, shape is [M, N] - :math:`W_{i}`: the i-th learned weight, shape is [M, N]
- :math:`out{i}`: the i-th element of out, shape is [batch_size, size].
- :math:`y^\mathrm{T}`: the transpose of :math:`y_{2}`. - :math:`y^\mathrm{T}`: the transpose of :math:`y_{2}`.
The simple usage is: The simple usage is:
...@@ -8079,8 +8079,8 @@ def bilinear_tensor_product(x, ...@@ -8079,8 +8079,8 @@ def bilinear_tensor_product(x,
tensor = bilinear_tensor_product(x=layer1, y=layer2, size=1000) tensor = bilinear_tensor_product(x=layer1, y=layer2, size=1000)
Args: Args:
x (Variable): 3-D input tensor with shape [N x M x P] x (Variable): 2-D input tensor with shape [batch_size, M]
y (Variable): 3-D input tensor with shape [N x M x P] y (Variable): 2-D input tensor with shape [batch_size, N]
size (int): The dimension of this layer. size (int): The dimension of this layer.
act (str, default None): Activation to be applied to the output of this layer. act (str, default None): Activation to be applied to the output of this layer.
name (str, default None): The name of this layer. name (str, default None): The name of this layer.
......
...@@ -901,6 +901,16 @@ class TestBook(unittest.TestCase): ...@@ -901,6 +901,16 @@ class TestBook(unittest.TestCase):
self.assertIsNotNone(data_1) self.assertIsNotNone(data_1)
print(str(program)) print(str(program))
def test_bilinear_tensor_product_layer(self):
program = Program()
with program_guard(program):
data = layers.data(name='data', shape=[4], dtype="float32")
theta = layers.data(name="theta", shape=[5], dtype="float32")
out = layers.bilinear_tensor_product(data, theta, 6)
print(str(program))
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册