RMSNorm

[cuda]RMSNorm核函数解析

### 计算原理 $RMSNorm = x * (sqrt(1/n * (x_i)^2 + eps)) * g$ ### torch实现 ```python class RMSNorm(torch.nn.Module): def __init__(self, dim: int, eps: float ......
函数 RMSNorm cuda
共1篇  :1/1页 首页上一页1下一页尾页