Skip to content

Commit ff0201c

Browse files
committed
Add max-pooling layer to the supported layers table
1 parent 49c65e4 commit ff0201c

File tree

1 file changed

+9
-7
lines changed

1 file changed

+9
-7
lines changed

README.md

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)