Tham số axes=2 trong numpy.tensordot() chỉ định co kép (double contraction): co 2 trục cuối của a với 2 trục đầu của b.
Ví dụ tương tự bài trước với axes=2
import numpy as np
arr1 = np.array(range(1, 9)).reshape(2, 2, 2)
arr2 = np.array(['p', 'q', 'r', 's'], dtype=object).reshape(2, 2)
print("Mảng 1:", arr1)
print("Mảng 2:", arr2)
# Co kép axes=2
result = np.tensordot(arr1, arr2, axes=2)
print("Tích tensor:", result)
Kết quả
Mảng 1: [[[1 2] [3 4]] [[5 6] [7 8]]]
Mảng 2: [['p' 'q'] ['r' 's']]
Tích tensor: ['pqqrrrssss' 'pppppqqqqqqrrrrrrrssssssss']
axes=2 tương đương axes=([1,2], [0,1]) cho mảng 3D và 2D: co trục 1,2 của arr1 với trục 0,1 của arr2.