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) – 表示稠密维度数量的整数

本页目录