Python PyTorch sparse_用法及代码示例

本文简要介绍python语言中 torch.nn.init.sparse_ 的用法。

用法:

torch.nn.init.sparse_(tensor, sparsity, std=0.01)

参数：

• tensor-一个n维torch.Tensor

• sparsity-每列中要设置为零的元素分数

• std-用于生成非零值的正态分布的标准差

将 2D 输入 Tensor 填充为稀疏矩阵，其中非零元素将从正态分布 \mathcal{N}(0, 0.01) 中提取，如 Deep learning via Hessian-free optimization - Martens, J. (2010) 中所述。

例子

>>> w = torch.empty(3, 5)
>>> nn.init.sparse_(w, sparsity=0.1)