Skip to content

Commit

Permalink
Update neural/README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
ChanLumerico authored Aug 13, 2024
1 parent 2abec57 commit 4c41abb
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions luma/neural/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -12,9 +12,9 @@ Deep learning models and neural network utilities of Luma

| Class | Input Shape | Output Shape |
| --- | --- | --- |
| Convolution1D | $(N,C_{in},W)$ | $(N,C_{out},W)$ |
| Convolution2D | $(N,C_{in},H,W)$ | $(N,C_{out},H,W)$ |
| Convolution3D | $(N,C_{in},D,H,W)$ | $(N,C_{out},D,H,W)$ |
| `Convolution1D` | $(N,C_{in},W)$ | $(N,C_{out},W)$ |
| `Convolution2D` | $(N,C_{in},H,W)$ | $(N,C_{out},H,W)$ |
| `Convolution3D` | $(N,C_{in},D,H,W)$ | $(N,C_{out},D,H,W)$ |

### Pooling

Expand Down

0 comments on commit 4c41abb

Please sign in to comment.