pytorch: работа над партией тензоров? - PullRequest
1 голос
/ 03 июля 2019

У меня есть партии изображений размером (B,C,H,W) (факел. тензор ) я хочу ненормализовать его с помощью тензора среднего / стандартного (например :[0.4,0.3,0.2]) размером (C,H,W)

Мой вопрос: есть ли какие-либо операции в Pytorch, которые могут делать это, например, dataset.map или tf.expand_dims в Tensorflow?

Будьте осторожны, это не формат изображения PIL

...