numpy—np.tensordot

np.tensordot

numpy.tensordot(a, b, axes=2)

描述

沿指定的轴计算张量点积

参数

a, b : array_like
张量
axes ：int or (2,) array_like
如果为整数，则按顺序，对a的最后N个轴和b的前N个轴求和。相应轴的尺寸必须匹配
如果为两个array，则为要求和的轴列表，第一个序列适用于a，第二个序列适用于b。
两个元素array_like必须具有相同的长度。元素array_like里的元素代表序列a、b要删除的轴（见案例2）
输出：输入张量的点积

Note

1.案例1

2.案例2