torch.nn.init.constant()的用法

torch.nn.init.constant(tensor, val)
用val的值填充输入的张量或变量

参数:
tensor – n维的torch.Tensor或autograd.Variable
val – 用来填充张量的值

posted @ 2021-03-15 20:33  吴莫愁258  阅读(3143)  评论(0编辑  收藏  举报