基本的 Pytorch 张量乘法和加法

Question

我只是意识到我缺乏一些非常基本的 pytorch 张量数学。 如何使用 pytorch 张量执行以下操作？

lab_rs = (lab_rs * [100, 255, 255] - [0, 128, 128])

这在 numpy 中运行良好。 这是一个形状为 (3, 512, 1024) 的图像，我想分别从每个颜色通道乘以和减去值

我用张量尝试同样的错误是：

TypeError: only integer tensors of a single element can be converted to an index

Answer 1

您需要确保所有操作数都可以广播到相同的维度：

lab_rs = lab_rs * torch.tensor([[[100]], [[255]], [[255.]]]) - torch.tensor([[[0]], [[128]], [[128.]]])