快捷方式

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

文档

查阅 PyTorch 全面的开发者文档

查看文档

教程

获取面向初学者和高级开发者的深度教程

查看教程

资源

查找开发资源并获得问题解答

查看资源