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]])