TP并行张量是如何切分的，又是如何合并的，按行切和按列切有何不同？

回答思路

1. 准确说明行切分与列切分的具体划分方式和维度
2. 解释切分后如何通过通信完成合并与结果还原
3. 对比行/列切分在数学运算（如矩阵乘）中的行为差异
4. 能结合常见框架（如Megatron）说明2D切分策略