公式上，假设输入张量为 $X$，权重矩阵为 $W$，分割后计算： $$ Y = f(W_1 \cdot X) \quad \text{在 GPU1 上} \ Z = g(W_2 \cdot Y) \quad \text{在 GPU2 上} $$ 其中 $f$ 和 $g$ 是激活函数，这避免了整个模型驻留在单一GPU显存中。公式优化： $$ C = A \times B \quad \text{其中