評價此頁

torch.Tensor.sparse_resize_#

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

self 稀疏張量 重塑為期望的大小以及稀疏和密集維度的數量。

注意

如果 self 中指定的元素數量為零,那麼 sizesparse_dimdense_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) – 密集維度的數量