davrot/pytutorial
1
0
Fork 0
mirror of https://github.com/davrot/pytutorial.git synced 2025-04-19 05:36:42 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update README.md

Browse source 
Signed-off-by: David Rotermund <54365609+davrot@users.noreply.github.com>
This commit is contained in:
David Rotermund 2024-01-02 17:15:19 +01:00 • committed by GitHub
parent 2285e996aa
commit 28286a98bb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 24 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
numpy/einstein/README.md
View file

@ -12,7 +12,7 @@ Questions to [David Rotermund](mailto:davrot@uni-bremen.de)
## Examples ## Examples
$$\sum_i a_{i,i}$$ $$b = \sum_i a_{i,i}$$
```python ```python
import numpy as np import numpy as np
@ -80,6 +80,29 @@ Output:
[ 4 9 14 19 24]] [ 4 9 14 19 24]]
``` ```
$$b = \sum_i a_i a_i$$
```python
import numpy as np
a = np.arange(0, 5)
print(a) # -> [0 1 2 3 4]
print(np.einsum("i,i", a, a)) # -> 30
```
$$c(i) = \sum_j a_{i,j} b_j$$
```python
import numpy as np
a = np.arange(0, 25).reshape(5, 5)
b = np.arange(0, 5)
print(a)
print()
print(b) # -> [0 1 2 3 4]
print(np.einsum("ij,j", a, b)) # -> [ 30 80 130 180 230]
```
## [numpy.einsum](https://numpy.org/doc/stable/reference/generated/numpy.einsum.html) ## [numpy.einsum](https://numpy.org/doc/stable/reference/generated/numpy.einsum.html)