Skip to content

Commit

Permalink
modify API nn.Bilinear's doc, test=develop
Browse files Browse the repository at this point in the history
  • Loading branch information
wanghuancoder committed Mar 26, 2021
1 parent 372ac08 commit d92a178
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions python/paddle/nn/layer/common.py
Original file line number Diff line number Diff line change
Expand Up @@ -578,15 +578,15 @@ class Bilinear(layers.Layer):
.. math::
out_{i} = x1 * W_{i} * {x2^\mathrm{T}}, i=0,1,...,size-1
out_{i} = x1 * W_{i} * {x2^\mathrm{T}}, i=0,1,...,out_features-1
out = out + b
In this formula:
- :math:`x1`: the first input contains in1_features elements, shape is [batch_size, in1_features].
- :math:`x2`: the second input contains in2_features elements, shape is [batch_size, in2_features].
- :math:`W_{i}`: the i-th learned weight, shape is [in1_features, in2_features], and learned weight's shape is [out_features, in1_features, in2_features].
- :math:`out_{i}`: the i-th element of out, shape is [batch_size, out_features].
- :math:`out_{i}`: the i-th element of out, shape is [batch_size], and out's shape is [batch_size, out_features].
- :math:`b`: the learned bias, shape is [1, out_features].
- :math:`x2^\mathrm{T}`: the transpose of :math:`x2`.
Expand Down

0 comments on commit d92a178

Please sign in to comment.