Fonction torch.square de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.squareC'est une fonction de PyTorch utilisée pour calculer le carré élément par élément.

Définition de fonction

torch.square(input, out=None)

Exemple d'utilisation

Exemple

import torch

# Calculer le carré élément par élément
x = torch.tensor([1.0, 2.0, 3.0])
result = torch.square(x)
print(result)

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions