Basi c Pytorch тензор умножения и сложения

Question

Я просто понимаю, что мне не хватает какой-то базовой c тензорной математики. Как мне сделать следующее с тензором 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.]]])