python constant_Python init.constant方法代码示例

本文整理汇总了Python中torch.nn.init.constant方法的典型用法代码示例。如果您正苦于以下问题：Python init.constant方法的具体用法？Python init.constant怎么用？Python init.constant使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在模块torch.nn.init的用法示例。

在下文中一共展示了init.constant方法的25个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于我们的系统推荐出更棒的Python代码示例。

示例1: conv

​点赞 7

​

# 需要导入模块: from torch.nn import init [as 别名]

# 或者: from torch.nn.init import constant [as 别名]

def conv(in_planes, out_planes, kernel_size=3, stride=1, dilation=1, bias=False, transposed=False):

if transposed:

layer = nn.ConvTranspose2d(in_planes, out_planes, kernel_size=kernel_size, stride=stride, padding=1, output_padding=1, dilation=dilation, bias=bias)

# Bilinear interpolation init

w = torch.Tensor(kernel_size, kernel_size)

centre = ke