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

Pytorch中的批量矩阵乘法-与输出维数的处理混淆

matrix-multiplication batch-processing pytorch vectorization python

singa1994 · 技术社区 · 6 年前

我有两个阵列:

A
B

阵列 A 包含一批RGB图像,形状:

[batch, Width, Height, 3]

B 包含对图像进行“类似变换”操作所需的系数,形状:

[batch, 4, 4, 3]

简单地说,对单个图像的操作是输出环境映射的乘法( normalMap * Coefficients ).

我想要的输出应该保持形状:

[批次,宽度,高度,3]

我试着用 torch.bmm 但是失败了。这有可能吗?

0 回复 | 直到 6 年前

prosti 5 年前

我想你需要计算一下Pythorch的工作原理

BxCxHxW : number of mini-batches, channels, height, width

格式,也可以使用 matmul ,因为 bmm 使用张量或ndim/dim/rank=3。

我知道你可能会在网上找到这个,但无论如何:

batch1 = torch.randn(10, 3, 20, 10)
batch2 = torch.randn(10, 3, 10, 30)
res = torch.matmul(batch1, batch2)
res.size() # torch.Size([10, 3, 20, 30])

推荐文章

Costinean Sebastian · 如何优化使用for循环处理大型数组的MATLAB脚本的执行时间?"

10 月前

Gyro · 为什么我的%xmm3寄存器在vbroadcastsd中使用第一个参数,而不是第四个参数?

1 年前

aeiou · 对日志表进行矢量化处理,以确定最新的可用性

1 年前

IchKenneDeinenNamen · 当将向量存储为类属性时,为什么NumPy的矢量化评估速度较慢?

1 年前

roshoka · 如何在不使用for循环的情况下将函数应用于林空间的每个元素

1 年前

Xavier · 变长矢量的矢量化和

1 年前

bbbb · 矢量化或在大型数据帧上使用多重处理[重复]

1 年前

Sterling Butters · NumPy 1D近邻中循环的避免

1 年前

AnthonyML · 修改2d numpy数组中分隔符后的行中的所有元素

1 年前

Tim · np.vectorize和relativelta返回“relativeltaonlydiffs datetime/date”

1 年前