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

Pytorch-高效元素乘法?

pytorch

raaj · 技术社区 · 5 年前

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

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

1 回复 | 直到 5 年前

Jatentaki 5 年前

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

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

推荐文章

Ze0ruso · 做模型。eval()在代码中的位置有什么关系?

2 年前

rkraaijveld · sklearn的Coef。线性回归为无

2 年前

Ellis Thompson · 填充一个numpym数组以满足所需的大小

2 年前

Adam · 如何在多个模型上并行运行PyTorch推理?

2 年前

quinlan · 是dgl。nn。皮托克。GraphConv聚合边缘信息?

2 年前

Ofek Nourian · 完全连通的简单自回归模型预测

2 年前

MrThiele1708 · 运行时错误:mat1和mat2形状不能相乘(5400x64和5400x64)

2 年前

Moodhi · 计算GEC的F分数

2 年前

john wang · matlab使用可调用的python函数,但需要输入

2 年前

odbhut.shei.chhele · 如何解决错误:RuntimeError:收到0项ancdata

2 年前