batch size的改变会涉及矩阵运算