davrot/pytutorial
1
0
Fork 0
Code Issues Pull requests 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 2023-12-27 16:50:02 +01:00 committed by GitHub
parent 5380cd0421
commit 1475ddfdbf
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 29 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
numpy/broadcasting/README.md
View file

@ -222,3 +222,32 @@ Output:
[0 1 2 3 4] [0 1 2 3 4]
[0 1 2 3 4]] [0 1 2 3 4]]
``` ```
## Manual broadcast : [numpy.broadcast_shapes](https://numpy.org/doc/stable/reference/generated/numpy.broadcast_shapes.html#numpy-broadcast-shapes)
{: .topic-optional}
This is an optional topic!
```python
numpy.broadcast_shapes(*args)
```
> Broadcast the input shapes into a single shape.
```python
import numpy as np
a = np.arange(0, 3).reshape(3, 1)
b = np.arange(0, 5).reshape(1, 5)
print(a.shape) # -> (3, 1)
print(b.shape) # -> (1, 5)
new_shape = np.broadcast_shapes(a.shape, b.shape)
print(new_shape) # -> (3, 5)
new_shape = np.broadcast_shapes((3, 1), (1, 5))
print(new_shape) # -> (3, 5)
```