Я хочу вычислить следующую операцию над матрицей:

import numpy as np

x = np.arange(9).reshape((3,3))
result = np.zeros((3,3,3))
for i in range(3):
    for j in range(3):
        for k in range(3):
            result[i,j,k] = x[j,i] * x[j,k]

Который дает

array([[[  0.,   0.,   0.],
        [  9.,  12.,  15.],
        [ 36.,  42.,  48.]],

       [[  0.,   1.,   2.],
        [ 12.,  16.,  20.],
        [ 42.,  49.,  56.]],

       [[  0.,   2.,   4.],
        [ 15.,  20.,  25.],
        [ 48.,  56.,  64.]]])

Как и ожидалось.

Вопрос

Как я могу выполнить этот расчет с тензорными произведениями (без циклов) с помощью numpy?

Редактировать

Если элементы X являются векторами, операция вместо этого:

result[i,j,k] = np.dot(x[j,i] , x[j,k])

Каким будет подходящий оператор numpy для этого расчета?