Implement concat layer

This commit is contained in:
2024-03-17 21:38:29 +01:00
parent 9aab7cdb10
commit aac0c3a826
9 changed files with 121 additions and 33 deletions

View File

@@ -11,7 +11,7 @@ Convolutional Neural Network inference library running on CUDA.
- [x] Conv2d layer
- [ ] Max pooling
- [ ] Average pooling
- [ ] Concat layer
- [x] Concat layer
- [x] Sigmoid activation
- [x] ReLU activation
- [x] Softmax activation