Skip to content

Commit 0929f49

Browse files
committed
Add layer summary table to the README
1 parent 8899583 commit 0929f49

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

README.md

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,14 @@ Read the paper [here](https://arxiv.org/abs/1902.06714).
2121
* Data-based parallelism
2222
* Several activation functions
2323

24+
### Available layer types
25+
26+
| Layer type | Constructor name | Rank of output array | Forward pass | Backward pass |
27+
|------------|------------------|----------------------|--------------|---------------|
28+
| Input | `input` | 1, 3 | n/a | n/a |
29+
| Dense (fully-connected) | `dense` | 1 |||
30+
| Convolutional (2-d) | `conv2d` | 3 |||
31+
2432
## Getting started
2533

2634
Get the code:

0 commit comments

Comments
 (0)