Update 17_foundations.ipynb

This commit is contained in:
SOVIETIC-BOSS88 2020-06-03 22:06:17 +02:00 committed by GitHub
parent 62ac21d085
commit e3aae2359e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1137,7 +1137,7 @@
"torch.einsum('bi,ij,bj->b', a, b, c)\n",
"```\n",
"\n",
"will return a vector of size `b` where the `k`-th coordinate is the sum of `a[k,i] b[i,j] c[k,j]`. This notation is particularly convenient when you have more dimensions because of batches. For example, if you have two batches of matrices and want compute the matrix product per batch, you would could this: \n",
"will return a vector of size `b` where the `k`-th coordinate is the sum of `a[k,i] b[i,j] c[k,j]`. This notation is particularly convenient when you have more dimensions because of batches. For example, if you have two batches of matrices and want to compute the matrix product per batch, you would could this: \n",
"\n",
"```python\n",
"torch.einsum('bik,bkj->bij', a, b)\n",