【发布时间】:2021-01-25 04:01:03
【问题描述】:
谁能解释一下如何在 python 中做两个张量的内积以获得一维数组。例如,我有两个大小为 (6,6,6,6,6) 和 (6,6,6,6) 的张量。我需要一个大小为 (6,1) 或 (1,6) 的一维数组。
【问题讨论】:
标签: python-3.x multidimensional-array tensor dot-product numpy-einsum
谁能解释一下如何在 python 中做两个张量的内积以获得一维数组。例如,我有两个大小为 (6,6,6,6,6) 和 (6,6,6,6) 的张量。我需要一个大小为 (6,1) 或 (1,6) 的一维数组。
【问题讨论】:
标签: python-3.x multidimensional-array tensor dot-product numpy-einsum
Numpy 有一个函数tensordot,查看一下:
https://numpy.org/doc/stable/reference/generated/numpy.tensordot.html
【讨论】: