Spec-Zone.ru › PyTorch 2

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

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.Tensor.to_dense.html

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API