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

Pytorch-高效元素乘法?

pytorch

raaj · 技术社区 · 6 年前

目前,我正在创建一个新的向量W,在这里我将[100x3]元素与[100x3]张量叠加,然后再进行元素相乘。

我需要这样做很多次,这是太慢和记忆密集的方式。有更好的办法吗?

1 回复 | 直到 6 年前

Jatentaki 6 年前

* )在PyTorch中已经是元素级的了。此外,它 broadcasts . 所以呢

import torch
xyz = torch.randn(100, 3)
w = torch.randn(100, 1)
multiplied = xyz * w

推荐文章

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

5 月前

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

1 年前

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

1 年前

JohnnyWang97 · getattr引起的有趣错误

1 年前

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

1 年前

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

1 年前

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

1 年前

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

1 年前

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

1 年前

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

1 年前