PyTorch torch.sign Funktion

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

torch.signEs ist eine Funktion in PyTorch, die verwendet wird, um das Vorzeichen elementweise zurückzugeben. Sie gibt -1 (negativ), 0 (null) oder 1 (positiv) zurück.

Funktionsdefinition

torch.sign(input, out=None)

Anwendungsbeispiele

Beispiel

import torch

# Vorzeichen elementweise zurückgeben
x = torch.tensor([-2.0, 0.0, 3.0])
Ergebnis = torch.sign(x)
print(result)

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

Andere Erweiterungen