python-3.x - python map如何与torch.tensor一起工作?
问题描述
我现在在 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'
请为我澄清一下谢谢
解决方案
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)
它会起作用的。