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)