Struct DropoutDescriptor

• Defined in File Descriptors.h

Inheritance Relationships

Base Type

• public at::native::Descriptor< cudnnDropoutStruct, &cudnnCreateDropoutDescriptor, &cudnnDestroyDropoutDescriptor > (Template Class Descriptor)

Struct Documentation

struct DropoutDescriptor : public at::native::Descriptor<cudnnDropoutStruct, &cudnnCreateDropoutDescriptor, &cudnnDestroyDropoutDescriptor>

Public Functions

inline void initialize_rng(cudnnHandle_t handle, float dropout, long long int seed, const TensorOptions &options)

inline void set(cudnnHandle_t handle, float dropout, at::Tensor state_)

inline void set_no_dropout(cudnnHandle_t handle)

Public Members

at::Tensor state