评价此页

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) – 密集维度的数量