Skip to content

Commit e9ddf32

Browse files
authored
[modifier] add unsqueeze (#400)
1 parent 13b460e commit e9ddf32

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tinynn/graph/modifier.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2537,6 +2537,7 @@ def register_mask(self, modifiers, importance, sparsity):
25372537
'view': ReshapeModifier,
25382538
"flatten": ReIndexModifier,
25392539
"squeeze": ReIndexModifier,
2540+
'unsqueeze': ReIndexModifier,
25402541
nn.Flatten: ReIndexModifier,
25412542
'reshape': ReshapeModifier,
25422543
'transpose': ReIndexModifier,

0 commit comments

Comments
 (0)