Skip to content

Commit db82a4f

Browse files
author
Sebastien M. Popoff
committed
correct layer list
1 parent 486e128 commit db82a4f

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

README.md

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,16 +15,19 @@ pip install complexPyTorch
1515
## Complex Valued Networks with PyTorch
1616

1717
Artificial neural networks are mainly used for treating data encoded in real values, such as digitized images or sounds.
18-
In such systems, using complex-valued tensor would be quite useless.
18+
In such systems, using complex-valued tensors would be quite useless.
1919
However, for physic related topics, in particular when dealing with wave propagation, using complex values is interesting as the physics typically has linear, hence more simple, behavior when considering complex fields.
2020
complexPyTorch is a simple implementation of complex-valued functions and modules using the high-level API of PyTorch.
2121
Following [[C. Trabelsi et al., International Conference on Learning Representations, (2018)](https://openreview.net/forum?id=H1T2hmZAb)], it allows the following layers and functions to be used with complex values:
2222
* Linear
2323
* Conv2d
24+
* ConvTranspose2d
2425
* MaxPool2d
26+
* AvgPool2d
2527
* Relu (ℂRelu)
2628
* Sigmoid
2729
* Tanh
30+
* Dropout2d
2831
* BatchNorm1d (Naive and Covariance approach)
2932
* BatchNorm2d (Naive and Covariance approach)
3033
* GRU/BN-GRU Cell

0 commit comments

Comments
 (0)