import torch.nn as nn
a = nn.LSTM(3,3)
for name,v in a.named_parameters():
print(name)
print结果:
weight_ih_l0
weight_hh_l0
bias_ih_l0
bias_hh_l0
import torch.nn as nn
a = nn.LSTM(3,3)
for name,v in a.named_parameters():
print(name)
print结果:
weight_ih_l0
weight_hh_l0
bias_ih_l0
bias_hh_l0