torch.nn.init.constant_()函数

参考   torch.nn.init - 云 社区 - 腾讯云

torch.nn.init.constant_(tensor, val)[source]

用值val填充向量。

参数：

• tensor – an n-dimensional torch.Tensor
• val – the value to fill the tensor with

例：

代码语言：javascript复制

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

腾讯云 init tensor torch

0 人点赞