Иногда каналы RGB размещаются в измерении 0, а иногда — в измерении 1. Но обобщение можно производить путем отсчета с конца: они всегда расположены в измерении –3, третьем с конца. Вычисление невзвешенного среднего можно записать следующим образом:
PyTorch. Освещая глубокое обучение
·
Лука Антига
