We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8899583 commit 0929f49Copy full SHA for 0929f49
README.md
@@ -21,6 +21,14 @@ Read the paper [here](https://arxiv.org/abs/1902.06714).
21
* Data-based parallelism
22
* Several activation functions
23
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
32
## Getting started
33
34
Get the code:
0 commit comments