Skip to content

Commit bb3dc1a

Browse files
author
Robert Muchsel
authored
README: Add mention of 'Flatten()' (#239)
1 parent 4e47412 commit bb3dc1a

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

README.md

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
# ADI MAX78000/MAX78002 Model Training and Synthesis
22

3-
April 13, 2022
3+
June 1, 2022
44

55
ADI’s MAX78000/MAX78002 project is comprised of five repositories:
66

@@ -1578,7 +1578,7 @@ Dropout modules such as `torch.nn.Dropout()` and `torch.nn.Dropout2d()` are auto
15781578
15791579
*Note: Using [batch normalization](#Batch-Normalization) in conjunction with dropout can sometimes degrade training results.*
15801580
1581-
#### view() and reshape()
1581+
#### view(), reshape() and Flatten
15821582
15831583
There are two supported cases for `view()` or `reshape()`.
15841584
@@ -1591,6 +1591,7 @@ There are two supported cases for `view()` or `reshape()`.
15911591
2. Conversion from 1D and 2D to Fully Connected (“flattening”): The batch dimension (first dimension) must stay the same, and the other dimensions are combined (i.e., M = C×H×W or M = C×L).
15921592
Example:
15931593
`x = x.view(x.size(0), -1) # Flatten`
1594+
An alternate way to express the flatten operation is `torch.nn.Flatten()`.
15941595
15951596
#### Support for Quantization
15961597

README.pdf

32.2 KB
Binary file not shown.

0 commit comments

Comments
 (0)