Add Unsqueeze Module

This module, nn.Unsqueeze, is the "inverse" operation of nn.Squeeze. It inserts a singleton dim (i.e., dimension 1) at specified position.

1 files changed, 1 insertions, 0 deletions

diff --git a/init.lua b/init.lua
index 96cbb53..5926c4f 100644
--- a/init.lua
+++ b/init.lua
@@ -27,6 +27,7 @@ require('nn.Select')
 require('nn.Narrow')
 require('nn.Index')
 require('nn.Squeeze')
+require('nn.Unsqueeze')
 require('nn.Replicate')
 require('nn.Transpose')
 require('nn.BatchNormalization')