CODEWITHSUNDEEP
pythonpytorchsparse-matrixtensortorch
Daniel Zamnius
Daniel Zamnius

Reputation: 83

How to convert a PyTorch sparse_coo_tensor into a PyTorch dense tensor?

I create a sparse_coo tensor in PyTorch:

import torch

# create indices
i = torch.tensor([[0, 1, 1],
                  [2, 0, 2]])
# create values
v = torch.tensor([3, 4, 5], dtype=torch.float32)

# create sparse_coo_tensor
sparse_tensor = torch.sparse_coo_tensor(i, v, [2, 4])

Now I want to convert a PyTorch sparse tensor into a PyTorch dense tensor. Which function can be used?

Upvotes: 1

Views: 5983

Answers (1)

trialNerror
trialNerror

Reputation: 3563

you can use to_dense as suggested in this example :

s = torch.sparse_coo_tensor(i, v, [2, 4])
s_dense = s.to_dense()

And by the way, the documentation is here

Upvotes: 2

Related Questions