Fonction torch.full de PyTorch

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

torch.fullest une fonction de PyTorch utilisée pour créer un tenseur rempli avec une valeur spécifiée.

Définition de fonction

torch.full(size, fill_value, dtype=None, device=None, requires_grad=False)

Exemple d'utilisation

Exemple

import torch

# Créer un tenseur 3x4 avec la valeur de remplissage 7
x = torch.full((3, 4), 7)

print(x)

Le résultat de sortie est:

tensor([[7., 7., 7., 7.],
        [7., 7., 7., 7.],
        [7., 7., 7., 7.]])

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

Autres extensions