File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -63,17 +63,17 @@ class Bilinear(Module):
6363 r"""Applies a bilinear transformation to the incoming data: :math:`y = x_1 * A * x_2 + b`
6464
6565 Args:
66- in_features1 : size of each first input sample
67- in_features2 : size of each second input sample
66+ in1_features : size of each first input sample
67+ in2_features : size of each second input sample
6868 out_features: size of each output sample
6969 bias: If set to False, the layer will not learn an additive bias. Default: True
7070
7171 Shape:
72- - Input: :math:`(N, in\_features1 )`, :math:`(N, in\_features2 )`
72+ - Input: :math:`(N, in1\_features )`, :math:`(N, in2\_features )`
7373 - Output: :math:`(N, out\_features)`
7474
7575 Attributes:
76- weight: the learnable weights of the module of shape (out_features x in_features1 x in_features2 )
76+ weight: the learnable weights of the module of shape (out_features x in1_features x in2_features )
7777 bias: the learnable bias of the module of shape (out_features)
7878
7979 Examples::
You can’t perform that action at this time.
0 commit comments