Fonction torch.sparse_coo_tensor de PyTorch

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

torch.sparse_coo_tensorest une fonction de PyTorch utilisée pour construire des tenseurs creux COO (format de coordonnées).

Définition de fonction

torch.sparse_coo_tensor(indices, values, size=None, dtype=None, device=None, requires_grad=False)

Exemple d'utilisation

Exemple

import torch

# Créer un tenseur creux COO
# Coordonnées d'index
indices = torch.tensor([[0, 1, 2], [0, 2, 1]])
# Valeurs correspondantes
values = torch.tensor([1.0, 2.0, 3.0])
# Taille du tenseur
size = (3, 3)

# Construire un tenseur creux COO
x = torch.sparse_coo_tensor(indices, values, size)

print(x)
print("Tenseur dense :")
print(x.to_dense())

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

Autres extensions