scipy.sparse.coo_matrix.

tensordot#

coo_matrix.tensordot(other, axes=2)[source]#

返回与另一个数组沿着给定轴的张量点积。

tensordot 不同于 dot 和 matmul,因为可以为第一个和第二个数组的每一个选择任何轴,并且就像矩阵乘法一样计算乘积之和,只是不仅仅是第一个数组的行乘以第二个数组的列。它采用沿指定轴的向量集合的点积。这里我们甚至可以根据需要计算沿着两个甚至更多轴的乘积之和。因此,tensordot 是应用于任何维度 >= 1 的数组的点积计算。它类似于 matmul,但适用于每个矩阵的任意轴。

给定两个张量,ab,以及所需的轴指定为一个包含两个轴号序列的 2 元组/列表/数组 (a_axes, b_axes),对 ab 的元素(分量)在由 a_axesb_axes 指定的轴上求乘积之和。 axes 输入可以是一个非负整数 N;如果是,则 a 的最后 N 维和 b 的前 N 维将被求和。

参数:
a, barray_like

要进行“点积”的张量。

axesint 或 (2,) array_like
  • integer_like 如果是整数 N,则按顺序对 a 的最后 N 个轴和 b 的前 N 个轴求和。 相应轴的大小必须匹配。

  • (2,) array_like 要进行求和的轴的序列的 2 元组,第一个应用于 a,第二个应用于 b。 序列的长度必须相同。 ab 之间相应轴的形状必须匹配。

返回值:
outputcoo_array

此数组与 other 的张量点积。 如果 other 是密集/稀疏的,它将是密集/稀疏的。

另请参阅

dot

示例

>>> import numpy as np
>>> import scipy.sparse
>>> A = scipy.sparse.coo_array([[[2, 3], [0, 0]], [[0, 1], [0, 5]]])
>>> A.shape
(2, 2, 2)

整数轴 N 是 (range(-N, 0), range(0, N)) 的简写

>>> A.tensordot(A, axes=1).toarray()
array([[[[ 4,  9],
         [ 0, 15]],

        [[ 0,  0],
         [ 0,  0]]],


       [[[ 0,  1],
         [ 0,  5]],

        [[ 0,  5],
         [ 0, 25]]]])
>>> A.tensordot(A, axes=2).toarray()
array([[ 4,  6],
       [ 0, 25]])
>>> A.tensordot(A, axes=3)
array(39)

使用元组作为轴

>>> a = scipy.sparse.coo_array(np.arange(60).reshape(3,4,5))
>>> b = np.arange(24).reshape(4,3,2)
>>> c = a.tensordot(b, axes=([1,0],[0,1]))
>>> c.shape
(5, 2)
>>> c
array([[4400, 4730],
       [4532, 4874],
       [4664, 5018],
       [4796, 5162],
       [4928, 5306]])