summary的计算存在错误

[li1127217ye]

对比了torchkeras与torchinfo的结果，发现torchkeras的计算存在错误。