Python PyTorch constant_用法及代碼示例

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

用法:

torch.nn.init.constant_(tensor, val)

參數：

• tensor-一個n維torch.Tensor

• val-填充張量的值

用值 \text{val} 填充輸入張量。

例子

>>> w = torch.empty(3, 5)
>>> nn.init.constant_(w, 0.3)