torch.dot does not broadcast
torch.dot() can only work for 1 dimension tensor. http://pytorch.org/docs/master/torch.html https://github.com/pytorch/pytorch/issues/2313
This commit is contained in:
@ -279,10 +279,11 @@
|
||||
"source": [
|
||||
"# incorrect method\n",
|
||||
"data = np.array(data)\n",
|
||||
"tensor = torch.Tensor([1,2,3,4]\n",
|
||||
"print(\n",
|
||||
" '\\nmatrix multiplication (dot)',\n",
|
||||
" '\\nnumpy: ', data.dot(data), # [[7, 10], [15, 22]]\n",
|
||||
" '\\ntorch: ', tensor.dot(tensor) # this will convert tensor to [1,2,3,4], you'll get 30.0\n",
|
||||
" '\\ntorch: ', torch.dot(tensor.dot(tensor) # 30.0. Beware that torch.dot does not broadcast, only works for 1-dimensional tensor\n",
|
||||
")"
|
||||
]
|
||||
},
|
||||
@ -360,7 +361,8 @@
|
||||
}
|
||||
],
|
||||
"source": [
|
||||
"tensor.dot(tensor)"
|
||||
"torch.dot(torch.Tensor([2, 3]), torch.Tensor([2, 1]))
|
||||
7.0"
|
||||
]
|
||||
},
|
||||
{
|
||||
|
||||
Reference in New Issue
Block a user