JZTXT
首页
Ai
Java
Python
Android
Mysql
JavaScript
Html
CSS
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
更新时间 2023-08-20
共1篇 :1/1页
首页
上一页
1
下一页
尾页