对标识 Layer 进行序列并行策略。
注解
被标识的 Layer 的输入张量的形状应该为 [b, s, h]。
[b, s, h]
代码示例
COPY-FROM: paddle.distributed.SequenceParallelEnable