tlx.split的pytorch版本实现错误

[zqwang-cn]

New Issue Checklist

• [x] I have read the Contribution Guidelines
• [x] I searched for existing GitHub issues

Issue Description

tlx.split的pytorch版本实现错误。tlx.split的第2个参数num_or_size_splits类型为int时，表示需要拆分成的tensor数量；但是在torch.split中第2个参数的意义为拆分出的每个tensor中在该维度上的长度。因此需要在调用前将该参数进行转换。

[hanjr92]

已修复 #26