Create Time: 24th August 2024
Title: Dropout
status: DONE
Author:
- AllenYGY
tags:
- NOTE
- DeepLearning
一个好的模型需要对输入数据的扰动鲁棒
Tikhonov
正则import torch
def dropout_layer(X,dropout):
assert 0<= dropout <=1 #概率值在0-1之间
if dropout == 1:
return torch.zeros_like(X)
if dropout == 0:
return X
mask = (torch.randn(X.shape) > dropout).float
return mask * X / (1.0-dropout)