pytorch 开根号函数

在PyTorch中，你可以使用内置的`torch.sqrt()`函数来计算数值的平方根。这是一个常用的数学运算，在处理深度学习模型中的向量或张量时非常实用。这个函数接受一个Tensor作为输入，并返回一个新的Tensor，其中元素是输入Tensor对应元素的平方根。

例如：

import torch

# 创建一个张量
x = torch.tensor([[4., 9.], [16., 25.]])
# 计算每个元素的平方根
sqrt_x = torch.sqrt(x)

运行这段代码后，`sqrt_x`将会是一个新的二维张量，其值分别是[2., 3., 4., 5.]。