Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Tích tensor cho mảng khác chiều trong Python

numpy.tensordot() mặc định (axes=2) thực hiện co tensor tiêu chuẩn: co các trục chung cuối của a và đầu của b.

Ví dụ mặc định

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)
print("Kích thước mảng 1:", arr1.shape)
print("Kích thước mảng 2:", arr2.shape)

# Tích tensor mặc định
result = np.tensordot(arr1, arr2)
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']]
Kích thước mảng 1: (2, 2, 2)
Kích thước mảng 2: (2, 2)
Tích tensor: ['pqqrrrssss' 'pppppqqqqqqrrrrrrrssssssss']

Mặc định axes=2 co 2 trục cuối của arr1 (2,2) với 2 trục của arr2 (2,2), kết quả 1D kích thước 2 (trục đầu của arr1).