Python PyTorch dirac_用法及代碼示例

本文簡要介紹python語言中 torch.nn.init.dirac_ 的用法。

用法:

torch.nn.init.dirac_(tensor, groups=1)

參數：

• tensor-{3, 4, 5} 維torch.Tensor

• groups(可選的) -卷積層中的組數(默認值：1)

使用 Dirac delta 函數填充 {3, 4, 5} 維輸入 Tensor。保留Convolutional 層中輸入的標識，其中盡可能多地保留輸入通道。在組>1的情況下，每組通道都保留身份

例子

>>> w = torch.empty(3, 16, 5, 5)
>>> nn.init.dirac_(w)
>>> w = torch.empty(3, 24, 5, 5)
>>> nn.init.dirac_(w, 3)