"输入"是一个关键字,当用作参数名称时会导致错误(在PyTorch中)？

Question

所以我有一行代码:

packed_embeddings = pack_padded_sequence(input=embeddings,
                                                 lengths=lengths,
                                                 batch_first=True)

这引发了我这个错误:

  File "/Users/kwj/anaconda3/lib/python3.6/site-packages/torch/onnx/__init__.py", line 130, in might_trace
    first_arg = args[0]

IndexError: tuple index out of range

但如果我拿出"输入",神奇地修复自己:

    packed_embeddings = pack_padded_sequence(embeddings,
                                             lengths=lengths,
                                             batch_first=True)

这是PyTorch文档中的函数规范:

https://pytorch.org/docs/stable/_modules/torch/nn/utils/rnn.html#pack_padded_sequence

我正在使用Python3和PyTorch 0.4.我错过了一些非常基本的东西吗 不确定这是我的问题,还是PyTorch特定的问题......在这里很困惑.

谢谢

Answer 1

这里发生的情况是，pack_padded_sequence被装饰为返回部分应用的函数，并且在装饰代码中，有一个函数接受参数为*args, **kwargs. 该函数传递args给另一个函数，该函数检查第一个arg. 当您将所有参数作为packed_padded_sequence关键字参数传递时，args为空，因此args[0]会引发IndexError. 如果您input作为位置参数传递，args则不为空，并且IndexError不会引发。

此示例代码演示了该行为（Pytorch代码不容易阅读）。

def decorator(func):
    def wrapper(*args, **kwargs):
        print('Args:', repr(args))
        print('Kwargs:', repr(kwargs))
        return func(*args, **kwargs)
    return wrapper   

@decorator
def f(a, b=0, c=0):
    return a, b, c


if __name__ == '__main__':
    print('Positional argument...')
    print(f(1, b=2, c=3))
    print('All keyword arguments...')
    print(f(a=1, b=2, c=3))

该代码产生以下输出：

Positional argument...
Args: (1,)                        <- Args is populated
Kwargs: {'b': 2, 'c': 3}
(1, 2, 3)
All keyword arguments...
Args: ()                          <- Args is empty
Kwargs: {'a': 1, 'b': 2, 'c': 3}
(1, 2, 3)