Created by: LutaoChu
Function optimization
OPs
1 optimize performance
2 add parameters "dtype", "name" 3 add dygraph sample 4 if axis=None, flatten the input