nn.Parameter 上的 Torch 操作

xiaoxingxing pytorch 186

原文标题Torch operation on nn.Parameter

我有一个参数列表,我想总结列表中的所有元素

import torch
from torch import nn

a = nn.Parameter(torch.rand(1))
b = nn.Parameter(torch.rand(1))

my_list = [a, b]
torch.sum(*my_list)

我收到错误

Traceback (most recent call last):
  File "<input>", line 8, in <module>
TypeError: sum() received an invalid combination of arguments - got (Parameter, Parameter), but expected one of:
 * (Tensor input, *, torch.dtype dtype)
 * (Tensor input, tuple of ints dim, bool keepdim, *, torch.dtype dtype, Tensor out)
 * (Tensor input, tuple of names dim, bool keepdim, *, torch.dtype dtype, Tensor out)

我想知道是否有办法执行像torch.sumonParameters这样的操作?

原文链接:https://stackoverflow.com//questions/71490306/torch-operation-on-nn-parameter

回复

我来回复
  • White_Sirilo的头像
    White_Sirilo 评论

    您可能想要访问一个示例,它是一个带有 a.data 的 nn.Parameter(),它返回一个火炬张量。我希望这会有所帮助。

    尝试打印 a 的 .data 字段以获得一些直觉:

    print(a.data)
    
    2年前 0条评论