Pycharm 抱怨将 [] 运算符与 pytorch 的 nn 一起使用。模块列表对象

Question

Pycharm的类型检查抱怨以下代码块(取自Pytorch的文档(：

import torch.nn as nn
class MyModule(nn.Module):
def __init__(self):
super(MyModule, self).__init__()
self.linears = nn.ModuleList([nn.Linear(10, 10) for i in range(10)])
def forward(self, x):
# ModuleList can act as an iterable, or be indexed using ints
for i, l in enumerate(self.linears):
x = self.linears[i // 2](x) + l(x)
#                   ^---------- complains here about:
# Cannot find reference '[' in 'input:(Any, ...), kwargs:dict) -> Any'
return x

有没有办法在不抑制的情况下解决此问题？

Answer 1

这是一个已知问题，https://youtrack.jetbrains.com/issue/PY-37601，将在 2020.1 中修复。