计算前始终将数组展平为一维向量

逐元素计算平方根，并返回形状相同的数组

原地将每个输入值替换为四舍五入后的整数平方根

对整个数组求和，只返回总和的平方根