Есть ли Python-эквивалент функции развертки R для многомерных массивов?

Question

Мне нужен эквивалент для функции развертки, но для работы с многомерными массивами, а не с фреймом данных

Например, когда вызывается следующий код:

x <- array(1, dim=2:4)
y <- array(c(1,2,3,4,5,6), dim=c(2,3))
sweep(x, MARGIN = c(1,2), y,FUN='*')

Результат:

, , 1

     [,1] [,2] [,3]
[1,]    1    3    5
[2,]    2    4    6

, , 2

     [,1] [,2] [,3]
[1,]    1    3    5
[2,]    2    4    6

, , 3

     [,1] [,2] [,3]
[1,]    1    3    5
[2,]    2    4    6

, , 4

     [,1] [,2] [,3]
[1,]    1    3    5
[2,]    2    4    6

Что эквивалентно свип-коду в Python? Есть ли встроенные функции, которые могут помочь?