摘要:
nn.Sequential是一个容器,用于按顺序放入神经网络的各个结构。 看一下构造函数,他会把每个模型放入列表中。 在前向传播中通过循环调用每个部分。 阅读全文
摘要:
def dropout(X,drop_prob):X=X.float()//将张量变成浮点数张量 assert 0<=drop_prob<=1//drop_prob不满足0-1则终止程序 keep_prob=1-drop_prob//对未丢弃的函数进行拉伸 if keep_prob==0: retu 阅读全文
摘要:
返回和输入矩阵形状相同的全0矩阵。相当于torch.zeros(x.size()) 阅读全文
摘要:
assert 1>=dropout>=0 如果assert后边条件为假,则终止程序运行。 阅读全文