Python PyTorch Tensor.to_dense用法及代码示例

本文简要介绍python语言中 torch.Tensor.to_dense 的用法。

用法:

Tensor.to_dense() → Tensor

创建 self 的跨步副本。

警告

如果 self 是跨步张量，则引发错误。

例子：

>>> s = torch.sparse_coo_tensor(
...        torch.tensor([[1, 1],
...                      [0, 2]]),
...        torch.tensor([9, 10]),
...        size=(3, 3))
>>> s.to_dense()
tensor([[ 0,  0,  0],
        [ 9,  0, 10],
        [ 0,  0,  0]])