实现 RMSNorm

为一维浮点张量实现 RMS（均方根）归一化。给定向量 x 和 epsilon，先计算向量上的 mean(x^2)，再令 scale = sqrt(mean(x^2) + eps)，最后返回 x / scale。该运算必须可微，因此请使用张量算子（torch.mean / torch.sqrt）完成，不要使用原地 Python 循环。