评价此页

torch.Tensor.sparse_resize_and_clear_#

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

将稀疏张量 self 中所有指定的元素移除,并将其重塑为期望的大小以及稀疏和稠密维度的数量。

参数
  • size (torch.Size) – 期望的大小。

  • sparse_dim (int) – 稀疏维度的数量

  • dense_dim (int) – 稠密维度的数量