1

我现在在 python 中，所以我试图从pytorch 教程中理解这一行。

x_train, y_train, x_valid, y_valid = map(
    torch.tensor, (x_train, y_train, x_valid, y_valid)
)

我了解 map 如何在单个元素上工作

def sqr(a):
    return a * a

a = [1, 2, 3, 4]    

a = map(sqr, a)
print(list(a))

在这里我需要使用list(a)将地图对象转换回列表。

但我不明白的是，它是如何对多个变量起作用的？

如果我尝试这样做

def sqr(a):
    return a * a


a = [1, 2, 3, 4]
b = [1, 3, 5, 7]

a, b = map(sqr, (a, b))
print(list(a))
print(list(b))

我收到一个错误：TypeError: can't multiply sequence by non-int of type 'list'

请为我澄清一下谢谢

4

1 回答 1

4

map它以与列表/列表元组相同的方式工作，它获取给定输入的元素，无论它是什么。

之所以torch.tensor有效，是因为它接受一个列表作为输入。

如果您展开您提供的以下行：

x_train, y_train, x_valid, y_valid = map(
    torch.tensor, (x_train, y_train, x_valid, y_valid)
)

这和这样做是一样的：

x_train, y_train, x_valid, y_valid = [torch.tensor(x_train), torch.tensor(y_train), torch.tensor(x_valid), torch.tensor(y_valid)]

另一方面，您的sqr函数不接受列表。a它期望一个标量类型为平方，而你的 an不是这种情况b，它们是列表。

但是，如果您更改sqr为：

def sqr(a):
    return [s * s for s in a]


a = [1, 2, 3, 4]
b = [1, 3, 5, 7]

a, b = map(sqr, (a, b))

或如@Jean 所建议，a, b = map(sqr, x) for x in (a, b)

它会起作用的。

于 2019-05-28T15:34:29.613 回答