LINUX.ORG.RU

numpy.tensordot поясните значение 3-го аргумента

 ,


0

1

Всем питон!
Поясните пожалуйста смысл третьего аргумента в функции numpy.tensordot. Тот, который axes. Да, читал документацию. Да, ничерта не понял. По каким измерениям там что суммируется. Особенно интересно, что происходит, когда в axes массив.

★★★
Ответ на: комментарий от Solace

В мане сказано, что по этим измерениям будет производиться какое-то там суммирование.

LIKAN ★★★
() автор топика
Ответ на: комментарий от Solace

Я так понимаю речь идет об этом Tensor Product (the single dot product) of two Tensors но куда тут воткнуть размерности, о которых идет речь в документации питона?

LIKAN ★★★
() автор топика

Посмотри ещё einsum.


In [14]: a = np.arange(60.).reshape(3, 4, 5)

In [15]: b = np.arange(24.).reshape(4, 3, 2)

In [16]: c = np.tensordot(a, b, axes=([1, 0], [0, 1]))

In [17]: d = np.einsum("ijm,jin->mn", a, b)

In [18]: np.all(d == c)
Out[18]: True

einsum делает $\sum_{ij} a_{ijm} b_{jin}$, а в tensordot задаются номера измерений, по которым суммируют, а не индексы

anonymous
()
Вы не можете добавлять комментарии в эту тему. Тема перемещена в архив.