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) – количество плотных измерений