def dropout_multiple(dropout_rate: float, rng: RandomStreams, use_dropout: bool, *parameters):
return tuple([dropout(dropout_rate, rng, p, use_dropout) for p in parameters])
微信
公众号
扫码关注公众号