Mindspore.Parameter求助
收藏回复举报
Mindspore.Parameter求助
t('forum.solved') 已解决
新人帖
发表于2024-01-23 01:21:20
0 查看

我正在将pytorch的:

self.init_ops_h = nn.Parameter(torch.zeros(state_size))

这一个函数转换为mindspore,我采用了以下的转换:

zeros = ops.Zeros() 

a = zeros(state_size) 

self.init_ops_h = ms.Parameter(a)

但是报错如下:(Parameter函数报错)

ValueError: For primitive[Zeros], the input number must be greater than or equal to 2, but got 1.

请问这是什么问题呢?

我要发帖子