pytorch
pytorchnn.Linearnn.Parameter
nn.Linear
'''
class torch.nn.Linear(in_features,out_features,bias=True)
weight=Parameter(torch.Tensor(out_features,in_features))
bias=Parameter(torch.Tensor(out_features))
'''
import torch
my_input=torch.randn(128,100)
my_laye
原创
2020-06-18 17:23:26 ·
131 阅读 ·
0 评论