zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-28 22:28:11 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add dropout2d (#250)

Browse Source
This commit is contained in:
Justin Deschenaux
2023-12-22 17:02:29 +01:00
committed by GitHub
parent 8385f93cea
commit e8deca84e0
4 changed files with 63 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
docs/src/python/nn/layers.rst
View File

@@ -27,3 +27,6 @@ Layers
MultiHeadAttention
Sequential
QuantizedLinear
Dropout
Dropout2d