Хм, спасибо. Немного уточню задачу, а то всё равно не вполне понимаю.
Есть тензор. Его размерность: [None,4,5,100].
None - это размер батча, на этапе инициализации сети это неизвестная величина.
4 - это тип переменной: матожидание x, матожидание y, std и множитель при гауссиане.
5 - это сколько гауссиан надо наложить на матрицу данного канала
100 - это количество каналов.
Соответственно, на выходе должна быть размерность:
[None,140,140,100], где 140х140 - размер изображения, а 100 - число каналов.
Я понимаю, как multivariate_normal(mean=mean, cov=cov) применить к тензору вида [матожидание x, матожидание y, std, множитель]. Но если размерность как у меня?.. Может, как-то хитро применить map или lambda функции?...