added optional last argument batchMode to nn.Reshape
Reshape(...[,batchMode]) doc