python-3.x - Y 和 *Y 之间有什么区别吗，其中 Y 是用作输入参数的列表？

Question

我正在使用 torch.tensor.repeat()

x = torch.tensor([[1, 2, 3], [4, 5, 6]])
period = x.size(1)
repeats = [1,2]
result = x.repeat(*repeats)

结果是

tensor([[1, 2, 3, 1, 2, 3],
    [4, 5, 6, 4, 5, 6]])

如果我得到如下结果

result = x.repeat(repeats)

结果是一样的

tensor([[1, 2, 3, 1, 2, 3],
    [4, 5, 6, 4, 5, 6]])

似乎 x.repeat(repeats) 和 x.repeat(*repeats) 工作相同。

这是否意味着，对于输入参数，例如 Y，我可以使用 Y 或 *Y

Answer 1

有点。如果 repeats 是 ints 的(list 或 tuple)，那么它是等价的。但总的来说，规则似乎是:

• 如果第一个参数是列表或元组，则将其视为重复。忽略所有其他参数。
• 否则，将完整的*args作为repeats

因此，如果您的repeats 很奇怪，例如repeats=((1,2),(3,4))，那么a.repeat(* repeats) 成功并且与 a.repeat(1, 2)==a.repeat((1, 2)) 和 a.repeat(repeats) 失败。

注意:这是基于我自己的测试的观察结果。唯一的官方文档是函数的定义类型，例如repeat(torch.Size or int...) 这在语义方面并不完全清楚。

---

您还可以得到这样的错误信息:

TypeError: repeat(): argument 'repeats' (position 1) must be tuple of ints, not tuple

当你传递花车时。一般来说，错误报告可能会更好。