Skip to content

Commit 8b1d572

Browse files
committed
fix minor docs
1 parent 75f1989 commit 8b1d572

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

torch/nn/modules/linear.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff 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::

0 commit comments

Comments
 (0)