Функция torch.square в PyTorch

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.square— это функция в PyTorch, используемая для поэлементного возведения в квадрат.

Определение функции

torch.square(input, out=None)

Пример использования

Пример

import torch

# Вычислить квадрат поэлементно
x = torch.tensor([1.0, 2.0, 3.0])
result = torch.square(x)
print(result)

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения