pytorch / 2 / generated / torch.tensor.sparse_resize_.html

torch.Tensor.sparse_resize_

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

Resizes self sparse tensor to the desired size and the number of sparse and dense dimensions.

Note

If the number of specified elements in self is zero, then size, sparse_dim, and dense_dim can be any size and positive integers such that len(size) == sparse_dim + dense_dim.

If self specifies one or more elements, however, then each dimension in size must not be smaller than the corresponding dimension of self, sparse_dim must equal the number of sparse dimensions in self, and dense_dim must equal the number of dense dimensions in self.

Warning

Throws an error if self is not a sparse tensor.

Parameters
  • size (torch.Size) – the desired size. If self is non-empty sparse tensor, the desired size cannot be smaller than the original size.
  • sparse_dim (int) – the number of sparse dimensions
  • dense_dim (int) – the number of dense dimensions

© 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