torch.Tensor.to_dense¶
- Tensor.to_dense(dtype=None, *, masked_grad=True) Tensor ¶
如果
self
不是一个带步长的张量,则创建self
的带步长副本,否则返回self
。- 关键字参数
{dtype} –
masked_grad (bool, 可选) – 如果设置为
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]])