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