Spec-Zone.ru › PyTorch 2

torch.Tensor.sparse_resize_

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

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

Примечание

Если количество указанных элементов в self равно нулю, то size, sparse_dim и dense_dim могут иметь любой размер и быть положительными целыми числами, такими что len(size) == sparse_dim + dense_dim.

Если self указывает один или несколько элементов, то каждое измерение в size не должно быть меньше соответствующего измерения self, sparse_dim должно быть равно числу разреженных измерений в self, и dense_dim должно быть равно числу плотных измерений в self.

Предупреждение

Возникает ошибка, если self не является разреженным тензором.

Параметры
  • size (torch.Size) – желаемый размер. Если self является непустым разреженным тензором, желаемый размер не может быть меньше исходного размера.
  • sparse_dim (int) – количество разреженных измерений
  • dense_dim (int) – количество плотных измерений

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

Spec-Zone.ru

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