代码之家 › 专栏 › 技术社区 › GoingMyWay

PyTorch,按元素应用不同的功能

pytorch

GoingMyWay · 技术社区 · 6 年前

我定义了这样的张量

t_shape = [4, 1]
data = torch.rand(t_shape)

我想对每一行应用不同的函数。

funcs = [lambda x: x+1, lambda x: x**2, lambda x: x-1, lambda x: x*2]  # each function for each row.

我可以用下面的代码来实现

d = torch.tensor([f(data[i]) for i, f in enumerate(funcs)])

如何使用PyTorch中定义的更高级API以适当的方式实现这一点?

0 回复 | 直到 6 年前

Wasi Ahmad 6 年前

我认为你的解决方案很好。但它不适用于任何张量形状。您可以按如下方式稍微修改解决方案。

t_shape = [4, 10, 10]
data = torch.rand(t_shape)

funcs = [lambda x: x+1, lambda x: x**2, lambda x: x-1, lambda x: x*2]

# only change the following 2 lines
d = [f(data[i]) for i, f in enumerate(funcs)]
d = torch.stack(d, dim=0)

推荐文章

Saffy · 如何在IterableDataset上应用最小最大缩放?

1 年前

sanjeev mk · 通过索引从Pytorch或Numpy 2D数组中快速删除多行的方法

2 年前

Anonymous · 如何为零维火炬张量赋值?

2 年前

JohnnyWang97 · getattr引起的有趣错误

2 年前

Kamugg · 在PyTorch中使用不同分辨率图像训练DeepLabV3的最佳实践

2 年前

Stocavista · 无法在python中将float 64转换为float 32

2 年前

efwefwefwefwefw wefwefwefwef · 如何在PyTorch Conv1d层中仅在一侧应用填充?

2 年前

Okhr · 运行时错误:CUDA错误:在带有GTX 1660 Super的Debian 12虚拟机上不支持此操作

2 年前

Fatemeh · 如何从使用nn训练和保存的模型加载检查点。DataParallel到不使用nn的模型上。DataParallel?

2 年前

Twenkid · 将GPT2 h5型号转换为割炬,以转换为ggml形状不匹配

2 年前