PyTorch torch.sqrt 関数

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

torch.sqrtこれは PyTorch でテンソルの要素ごとの平方根を計算するための関数です。

関数定義

torch.sqrt(input, out=None)

使用例

例

import torch

x = torch.tensor([4.0, 9.0, 16.0, 25.0])

y = torch.sqrt(x)
print(y)

出力結果は：

tensor([2., 3., 4., 5.])

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

その他の拡張