File tree Expand file tree Collapse file tree 1 file changed +9
-7
lines changed Expand file tree Collapse file tree 1 file changed +9
-7
lines changed Original file line number Diff line number Diff line change @@ -16,18 +16,20 @@ Read the paper [here](https://arxiv.org/abs/1902.06714).
1616
1717## Features
1818
19- * Dense, fully connected neural networks of arbitrary shape and size
20- * Backprop with Mean Square Error cost function
19+ * Dense, fully connected neural layers
20+ * Convolutional and max-pooling layers (experimental, forward propagation only)
21+ * Stochastic and mini-batch gradient descent for back-propagation
2122* Data-based parallelism
2223* Several activation functions
2324
2425### Available layer types
2526
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 | ✅ | ❌ |
27+ | Layer type | Constructor name | Supported input layers | Rank of output array | Forward pass | Backward pass |
28+ | ------------| ------------------| ------------------------| ----------------------| --------------| ---------------|
29+ | Input (1-d and 3-d) | ` input ` | n/a | 1, 3 | n/a | n/a |
30+ | Dense (fully-connected) | ` input ` (1-d) | ` dense ` | 1 | ✅ | ✅ |
31+ | Convolutional (2-d) | ` input ` (3-d), ` conv2d ` , ` maxpool2d ` | ` conv2d ` | 3 | ✅ | ❌ |
32+ | Max-pooling (2-d) | ` input ` (3-d), ` conv2d ` , ` maxpool2d ` | ` maxpool2d ` | 3 | ✅ | ❌ |
3133
3234## Getting started
3335
You can’t perform that action at this time.
0 commit comments