Función PyTorch torch.sparse_csr_tensor
Manual de referencia de PyTorch torch
torch.sparse_csr_tensorEs una función de PyTorch para construir tensores dispersos CSR (filas dispersas comprimidas).
Definición de la función
torch.sparse_csr_tensor(crow_indices, col_indices, values, size=None, dtype=None, device=None, requires_grad=False)
Ejemplo de uso
Ejemplo
import torch
# Crear tensor disperso CSR
# Punteros de fila
crow_indices = torch.tensor([0, 2, 3])
# Índices de columna
col_indices = torch.tensor([0, 1, 1])
# Valores
values = torch.tensor([1.0, 2.0, 3.0])
# Tamaño del tensor
size = (2, 3)
# Construir tensor disperso CSR
x = torch.sparse_csr_tensor(crow_indices, col_indices, values, size)
print(x)
print("Tensor denso:")
print(x.to_dense())
# Crear tensor disperso CSR
# Punteros de fila
crow_indices = torch.tensor([0, 2, 3])
# Índices de columna
col_indices = torch.tensor([0, 1, 1])
# Valores
values = torch.tensor([1.0, 2.0, 3.0])
# Tamaño del tensor
size = (2, 3)
# Construir tensor disperso CSR
x = torch.sparse_csr_tensor(crow_indices, col_indices, values, size)
print(x)
print("Tensor denso:")
print(x.to_dense())
Otras extensiones