масштабировать массив [x, y, z] с весами в форме [x, y, 1] вдоль z

Question

TA.shape = (x, y, z) TB.shape = (x, y, 1)

func (TA, TB): масштабировать значение TA с TB вдоль оси z, возвращает (x, y, z) образный массив.

Как это сделать в клочья, не дублируя канал ТБ на (x, y, z)? Надеюсь, вопрос был ясен ~