快捷键

torch.Tensor.to_dense(分号)

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

如果 self 不是带偏移量的张量,则创建 self 的带偏移量副本,否则返回 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]])

© 版权所有 PyTorch 贡献者。

使用 Sphinx 构建,并使用 Read the Docs 提供的主题。

文档

PyTorch 的全面开发者文档

查看文档

教程

深入了解初学者和高级开发者的教程

查看教程

资源

查找开发资源并获得您的疑问解答

查看资源