美文网首页
关于torch.nn.Linear的权重shape一点疑问

关于torch.nn.Linear的权重shape一点疑问

作者: 点指兵兵_ | 来源:发表于2020-08-06 19:21 被阅读0次
import torch
a = torch.randn(60, 30)  # 
b = torch.nn.Linear(30, 15)  # 但是b的weight的shape是[15,30]
output = b(a) #猜测是Linear会对权重做转置
print('b.weight.shape:\n ', b.weight.shape)
print('b.bias.shape:\n', b.bias.shape)
print('output.shape:\n', output.shape)


ans = torch.mm(a, b.weight.t()) + b.bias #这里也是将b转置才可以计算,结果和上面是一致的. 
print('ans.shape:\n', ans.shape)

print(torch.equal(ans, output))

相关文章

  • 关于torch.nn.Linear的权重shape一点疑问

  • 什么是“抖音”账号权重

    关于账号权重,简单来讲就是你账号的优质程度!说直白一点,也就是你在抖音爸爸这里的位置,权重会影响你作品的...

  • Keras Layer自定义层

    build(input_shape):这是定义权重的方法,可训练的权应该在这里被加入列表self.trainabl...

  • 关于权重

    上为权重叠加示例,网页内容显示为蓝色 继承的权重为0:如下 此网页内容显示为红色 解析:p自己规定了颜色就用自己的...

  • 正则化

    1 TensorFlow优化模型之正则化疑问: 为什么复杂度和模型的权重有关, 通过限制 权重的大小来正则化 loss

  • 关于shape标签

    六个标签一一说明:基本属性(corners、gradient、padding、size、solid、stroke)...

  • 有一点关于婚姻的疑问

    今天 是2019年1月5日,距离我结婚典礼还有15天。 为什么会突然想写点东西,在2个月不长不短的备婚过...

  • pytorch 维度计算

    维度计算 关于 shape 和 size() 的区别, 二者是一样的,区别在于 shape 是为了和numpy对应...

  • webAudio和domAudio的区别(部分移动浏览器内,Au

    关于webAudio和domAudio的一点疑问原文https://forum.cocos.org/t/webau...

  • 关于“权重”的困惑

    昨天逛简书,在一篇文章的评论区,看到这样的留言:好想降下权重,为你点赞。 看到这句话,我当时就好奇了。点赞跟权重有...

网友评论

      本文标题:关于torch.nn.Linear的权重shape一点疑问

      本文链接:https://www.haomeiwen.com/subject/yrjnrktx.html