pytorch 平方和开根号

PyTorch 是一个开源的机器学习库，主要用于深度学习和计算机视觉领域。在 PyTorch 中计算一个张量（tensor）的平方和开根号可以通过内置函数方便地完成。

1. 计算平方：PyTorch 提供了 `torch.square()` 函数来计算输入张量的每个元素的平方。例如，如果有张量 `a`，则可以通过 `torch.square(a)` 来得到它的平方张量。

2. 计算开根号：PyTorch 提供了 `torch.sqrt()` 函数来计算输入张量每个元素的平方根。使用这个函数，可以通过 `torch.sqrt(a)` 来得到张量 `a` 中每个元素的平方根。

以下是使用这两个函数的简单示例：

import torch

# 创建一个张量
a = torch.tensor([1.0, 4.0, 9.0])

# 计算平方
squared = torch.square(a)
print("平方后的张量:", squared)

# 计算开根号
sqrt_a = torch.sqrt(a)
print("开根号后的张量:", sqrt_a)

输出将会是：

平方后的张量: tensor([ 1., 16., 81.])
开根号后的张量: tensor([1.0000, 2.0000, 3.0000])