當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

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 填充為稀疏矩陣，其中非零元素將從正態分布 中提取，如 Deep learning via Hessian-free optimization - Martens, J. (2010) 中所述。

例子

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

相關用法


注：本文由純淨天空篩選整理自pytorch.org大神的英文原創作品 torch.nn.init.sparse_。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。