代码拉取完成,页面将自动刷新
torch.nn.init.constant_(
tensor,
val
)
更多内容详见torch.nn.init.constant_。
class mindspore.common.initializer.Constant(value)(arr)
更多内容详见mindspore.common.initializer.Constant。
PyTorch:以常量val
填充输入的tensor。
MindSpore:以value
(整型或numpy数组)填充输入的numpy数组,且是原地更新输入值。
import mindspore
import torch
import numpy as np
# In MindSpore, fill a constant array with value(int or numpy array).
input_constant = np.array([1, 2, 3])
constant_init = mindspore.common.initializer.Constant(value=1)
out_constant = constant_init(input_constant)
print(input_constant)
# Out:
# [1 1 1]
# In torch, fill in the input tensor with constant val.
input_constant = np.array([1, 2, 3])
out_constant = torch.nn.init.constant_(
tensor=torch.tensor(input_constant),
val=1
)
print(out_constant)
# Out:
# tensor([1, 1, 1])
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。