У меня есть тензор формы [batch size, height, width, 1] изображений в градациях серого. Прежде всего, мне нужно рассчитать среднее значение и значения стандартного отклонения для всех изображений в пакете (и сохранить эти значения для дальнейшего использования). Каковы правильные параметры оси в этом случае?
Затем мне нужно выполнить среднее вычитание и нормализацию, также что-то вроде X -= mean и X /= std.
tf.image.per_image_standardization не вариант, поскольку он работает только для одного изображения.