Shortcuts

torch.Tensor.sparse_resize_

Tensor.sparse_resize_(size, sparse_dim, dense_dim) 张量

self 稀疏张量 调整为所需的大小和稀疏与密集维度的数量。

注意

如果self中指定元素的数量为零,那么 sizesparse_dimdense_dim可以是任意大小和正整数,使得len(size) == sparse_dim + dense_dim

如果 self 指定了一个或多个元素,那么 size 中的每个维度必须不小于 self 的相应维度,sparse_dim 必须等于 self 中的稀疏维度数量,并且 dense_dim 必须等于 self 中的稠密维度数量。

警告

如果 self 不是稀疏张量,则抛出错误。

Parameters
  • 大小 (torch.Size) – 所需的大小。如果 self 是非空稀疏张量,所需的大小不能小于原始大小。

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

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

优云智算