Ctrl+K
scipy.sparse.coo_matrix.

tensordot#

coo_matrix.tensordot(other, axes=2)[源代码]#

返回与其他数组沿给定轴的张量点积。

张量点积与点积和matmul的不同之处在于，可以为第一个和第二个数组的每一个选择任意轴，并且像矩阵乘法一样计算乘积之和，而不仅仅是第一个的行乘以第二个的列。它沿着指定的轴取向量集合的点积。这里，如果需要，我们甚至可以沿两个或多个轴取乘积之和。因此，张量点积是应用于任何维度>= 1 的数组的点积计算。它类似于matmul，但针对每个矩阵的任意轴。

给定两个张量 a 和 b，以及期望的轴，指定为包含两个轴编号序列的 2 元组/列表/数组 (a_axes, b_axes)，将 a 和 b 的元素（分量）在 a_axes 和 b_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。 这些序列的长度必须相同。 a 和 b 之间对应轴的形状必须匹配。

返回:
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]])
在本页上