torch.Tensor.to_dense

Tensor.to_dense(dtype=None, *, masked_grad=True) → Tensor

如果 self 不是 strided 張量，則建立其 strided 副本，否則返回 self。

關鍵字引數

• {dtype} –

• masked_grad (布林型別, 可選) – 如果設定為 True（預設值），並且 self 具有稀疏佈局，則 to_dense() 的反向傳播返回 grad.sparse_mask(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]])