We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5ce4a4a commit 8074180Copy full SHA for 8074180
torch/nn/modules/instancenorm.py
@@ -71,13 +71,13 @@ class InstanceNorm1d(_InstanceNorm):
71
>>> m = nn.InstanceNorm1d(100)
72
>>> # With Learnable Parameters
73
>>> m = nn.InstanceNorm1d(100, affine=True)
74
- >>> input = autograd.Variable(torch.randn(20, 100))
+ >>> input = autograd.Variable(torch.randn(20, 100, 40))
75
>>> output = m(input)
76
"""
77
78
def _check_input_dim(self, input):
79
if input.dim() != 3:
80
- raise ValueError('expected 2D or 3D input (got {}D input)'
+ raise ValueError('expected 3D input (got {}D input)'
81
.format(input.dim()))
82
super(InstanceNorm1d, self)._check_input_dim(input)
83
0 commit comments