PyTorch torch.randint-Funktion

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

torch.randintEs ist eine Funktion in PyTorch, mit der zufällige Ganzzahl-Tensoren erstellt werden.

Funktionsdefinition

torch.randint(low, high, size, dtype, device, requires_grad)

Anwendungsbeispiele

Beispiel

import torch

# Erzeuge einen zufälligen Ganzzahl-Tensor von 0 bis 9
x = torch.randint(0, 10, (3, 4))

print(x)

Das Ausgabeergebnis ist:

tensor([[4, 5, 2, 6],
        [7, 9, 3, 1],
        [2, 8, 5, 4]])

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

Andere Erweiterungen