Shortcuts

Struct DropoutDescriptor

Inheritance Relationships

Base Type

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

Struct Documentation

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

Public Functions

void initialize_rng(cudnnHandle_t handle, float dropout, long long int seed, const TensorOptions &options)
void set(cudnnHandle_t handle, float dropout, at::Tensor state_)
void set_no_dropout(cudnnHandle_t handle)

Public Members

at::Tensor state

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources