torch.Tensor.sparse_resize_and_clear_

Tensor.sparse_resize_and_clear_(size, sparse_dim, dense_dim) → Tensor

Удаляет все указанные элементы из разреженного тензора self и изменяет self до требуемого размера и количества разреженных и плотных измерений.

Параметры:

• size (torch.Size) – требуемый размер.
• sparse_dim (int) – количество разреженных измерений
• dense_dim (int) – количество плотных измерений