原始计算：Y = X @ W
分割后：Y = X @ [W1, W2, W3, W4]  # 4个GPU
每个GPU计算：Yi = X @ Wi
最后聚合：Y = [Y1, Y2, Y3, Y4]