Function at::_cudnn_init_dropout_state(double, bool, int64_t, at::TensorOptions)
Defined in File Functions.h
Function Documentation
-
inline at::Tensor at::_cudnn_init_dropout_state(double dropout, bool train, int64_t dropout_seed, at::TensorOptions options)