numpy—np.tensordot

文章目录

      • np.tensordot
        • 1.案例1
        • 2.案例2

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
numpy—np.tensordot_第1张图片

1.案例1

numpy—np.tensordot_第2张图片

2.案例2

numpy—np.tensordot_第3张图片
在这里插入图片描述

你可能感兴趣的:(#,numpy,tensordot)