From the tutorial, I got that 
Snipaste_2018-04-12_09-38-49.png1435×809 173 KB

we need to consider the batch_size when using the Linear layer. Then if I define own module ,do i need to consider the batch_size, e.g if the original input is a  input_feature  Tensor, if we consider the batch_size, then the input will be batch_size*input_feature Tensor