张量的内积

问题描述

有人可以解释一下如何在python中做两个张量的内积以获得一维数组。例如，我有两个张量分别为（6,6,6）和（6,6）的张量。我需要一个尺寸为（6,1）或（1,6）的一维数组。

解决方法

Numpy具有功能tensordot，请检查一下：
https://numpy.org/doc/stable/reference/generated/numpy.tensordot.html