torch.Tensor.sparse_resize_

Tensor.sparse_resize_(size, sparse_dim, dense_dim) → Tensor

将 self 稀疏张量 调整为所需的尺寸和稀疏维度和稠密维度的数量。

注意

如果 self 中指定元素的数量为零，则 size、sparse_dim 和 dense_dim 可以是任何尺寸和正整数，只要 len(size) == sparse_dim + dense_dim。

但是，如果 self 指定了一个或多个元素，则 size 中的每个维度不能小于 self 的对应维度，sparse_dim 必须等于 self 中稀疏维度的数量，dense_dim 必须等于 self 中稠密维度的数量。

警告

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

参数

• size (torch.Size) – 目标尺寸。如果 self 是非空的稀疏张量，则目标尺寸不能小于原始尺寸。

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

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