目录
【pytorch】self.modules() 和 self.children()的区别
【pytorch】torch.distributions.Normal()
【pytorch】torch.distributions.sample()
【pytorch】torch.distributions.rsample()
【pytorch】torch.distributions.log_prob(value)
【pytorch】self.modules() 和 self.children()的区别
self.modue和self.children的区别与联系
import torch
from torch import nn
# hyper parameters
in_dim=1
n_hidden_1=1
n_hidden_2=1
out_dim=1
class Net(nn.Module):
def __init__(self, in_dim, n_hidden_1, n_hidden_2, out_dim):
super().__init__()