[英]How to convert a PyTorch sparse_coo_tensor into a PyTorch dense tensor?
我在 PyTorch 中创建了一个 sparse_coo 张量:
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])
现在我想将 PyTorch 稀疏张量转换为 PyTorch 密集张量。 哪个function可以用?
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.