от
В numpy / PyTorch у меня есть две матрицы, например
X=[[1,2],[3,4],[5,6]]
,
Y=[[1,1],[2,2]]
. Я хотел бы расставить точки на каждой строке X с каждой строкой Y и получить результаты
[[3, 6],[7, 14], [11,22]]
Как мне этого добиться? Спасибо!              

Пожалуйста, войдите или зарегистрируйтесь для публикации ответа на этот вопрос.

3 Ответы

0 голосов
от
Я думаю, это то, что вы ищете:
import numpy as np

x= [[1,2],[3,4],[5,6]] 
y= [[1,1],[2,2]]

x = np.asarray(x) #convert list to numpy array 
y = np.asarray(y) #convert list to numpy array

product = np.dot(x, y.T)
.T
транспонирует матрицу, которая в данном случае необходима для умножения (из-за способа определения продуктов точек).
print(product)
выведет:
[[ 3  6]
 [ 7 14]
 [11 22]]
    
0 голосов
от
Использование
einsum
np.einsum('ij,kj



array([[ 3,  6],
       [ 7, 14],
       [11, 22]])
    
0 голосов
от
В
PyTorch
вы можете добиться этого, используя
torch.mm(a, b)
или
torch.matmul(a, b)
, как показано ниже:
x = np.array([[1,2],[3,4],[5,6]])
y = np.array([[1,1],[2,2]])
x = torch.from_numpy(x)
y = torch.from_numpy(y)
# print(torch.matmul(x, torch.t(y)))
print(torch.mm(x, torch.t(y)))
выход:
tensor([[ 3,  6],
        [ 7, 14],
        [11, 22]], dtype=torch.int32)
    
...