У меня есть два массива
A = np.array([[1,2,3],
[4,5,6]])
E = np.array([[0,0,0],
[1,1,1],
[2,2,2]])
, и я пытаюсь создать следующий:
np.array([ [[ 1, 1, 1],
[ 1, 2, 3],
[ 1, 4, 9]],
[[ 1, 1, 1],
[ 4, 5, 6],
[16, 25, 36]] ])
Я думал, что
np.power(A, E)
подойдетработа, но это приводит меня к ошибке, что операнды не могут быть переданы вместе.
С другой стороны,
np.power(A[0], E)
работает нормально, но выдает только первый элемент желаемогомассив.
Есть ли способ, которым я могу сделать этот вид "дистрибутивного возведения в степень" в векторизованном виде?