代码之家 › 专栏 › 技术社区 › 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])

推荐文章

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

7 月前

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 年前