Shortcuts

Class TensorDescriptor

Inheritance Relationships

Base Type

  • public at::native::Descriptor< cudnnTensorStruct,&cudnnCreateTensorDescriptor,&cudnnDestroyTensorDescriptor > (Template Class Descriptor)

Class Documentation

class at::native::TensorDescriptor : public at::native::Descriptor<cudnnTensorStruct, &cudnnCreateTensorDescriptor, &cudnnDestroyTensorDescriptor>

Public Functions

TensorDescriptor()
TensorDescriptor(const at::Tensor &t, size_t pad = 0)
void set(const at::Tensor &t, size_t pad = 0)
void set(cudnnDataType_t dataType, IntArrayRef sizes, IntArrayRef strides, size_t pad = 0)
void print()

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