List of all items
Structs
- CModule
- COptimizer
- NewAxis
- NoGradGuard
- Scalar
- Tensor
- TrainableCModule
- data::Iter2
- data::TextData
- data::TextDataIter
- display::PrinterOptions
- index::NewAxis
- jit::CModule
- jit::Object
- jit::TrainableCModule
- nn::Adam
- nn::AdamW
- nn::BatchNorm
- nn::BatchNormConfig
- nn::Conv
- nn::ConvConfigND
- nn::ConvTransposeConfigND
- nn::ConvTransposeND
- nn::Embedding
- nn::EmbeddingConfig
- nn::Func
- nn::FuncT
- nn::GRU
- nn::GRUState
- nn::GroupNorm
- nn::GroupNormConfig
- nn::Id
- nn::LSTM
- nn::LSTMState
- nn::LayerNorm
- nn::LayerNormConfig
- nn::Linear
- nn::LinearConfig
- nn::Optimizer
- nn::Path
- nn::RNNConfig
- nn::RmsProp
- nn::Sequential
- nn::SequentialT
- nn::Sgd
- nn::VarStore
- nn::Variables
- vision::dataset::Dataset
- vision::dinov2::DinoVisionTransformer
- vision::efficientnet::MBConvConfig
Enums
- Cuda
- Device
- IValue
- Kind
- Layout
- QEngine
- Reduction
- TchError
- TensorIndexer
- index::TensorIndexer
- jit::IValue
- kind::Kind
- nn::PaddingMode
- nn::init::FanInOut
- nn::init::Init
- nn::init::NonLinearity
- nn::init::NormalOrUniform
- utils::QEngine
Traits
Functions
- autocast
- display::set_print_options
- display::set_print_options_default
- display::set_print_options_full
- display::set_print_options_short
- get_num_interop_threads
- get_num_threads
- jit::f_fuser_cuda_is_enabled
- jit::f_fuser_cuda_set_enabled
- jit::f_get_profiling_mode
- jit::f_get_tensor_expr_fuser_enabled
- jit::f_set_graph_executor_optimize
- jit::f_set_profiling_mode
- jit::f_set_tensor_expr_fuser_enabled
- jit::fuser_cuda_is_enabled
- jit::fuser_cuda_set_enabled
- jit::get_profiling_mode
- jit::get_tensor_expr_fuser_enabled
- jit::set_graph_executor_optimize
- jit::set_profiling_mode
- jit::set_tensor_expr_fuser_enabled
- manual_seed
- maybe_init_cuda
- nn::adam
- nn::adamw
- nn::batch_norm1d
- nn::batch_norm2d
- nn::batch_norm3d
- nn::conv
- nn::conv1d
- nn::conv2d
- nn::conv3d
- nn::conv_transpose1d
- nn::conv_transpose2d
- nn::conv_transpose3d
- nn::embedding
- nn::func
- nn::func_t
- nn::group_norm
- nn::gru
- nn::init::f_init
- nn::init::init
- nn::layer_norm
- nn::linear
- nn::lstm
- nn::no_bias
- nn::rms_prop
- nn::seq
- nn::seq_t
- nn::sgd
- no_grad
- no_grad_guard
- set_num_interop_threads
- set_num_threads
- utils::get_num_interop_threads
- utils::get_num_threads
- utils::has_cuda
- utils::has_cudart
- utils::has_cusolver
- utils::has_hip
- utils::has_ipu
- utils::has_lapack
- utils::has_lazy
- utils::has_magma
- utils::has_mkl
- utils::has_mkldnn
- utils::has_mps
- utils::has_openmp
- utils::has_ort
- utils::has_vulkan
- utils::has_xla
- utils::manual_seed
- utils::set_num_interop_threads
- utils::set_num_threads
- utils::version_cudart
- utils::version_cudnn
- vision::alexnet::alexnet
- vision::cifar::load_dir
- vision::convmixer::c1024_20
- vision::convmixer::c1536_20
- vision::dataset::augmentation
- vision::dataset::random_crop
- vision::dataset::random_cutout
- vision::dataset::random_flip
- vision::densenet::densenet121
- vision::densenet::densenet161
- vision::densenet::densenet169
- vision::densenet::densenet201
- vision::dinov2::vit_small
- vision::efficientnet::b0
- vision::efficientnet::b1
- vision::efficientnet::b2
- vision::efficientnet::b3
- vision::efficientnet::b4
- vision::efficientnet::b5
- vision::efficientnet::b6
- vision::efficientnet::b7
- vision::efficientnet::conv2d_same
- vision::image::load
- vision::image::load_and_resize
- vision::image::load_and_resize_from_memory
- vision::image::load_dir
- vision::image::load_from_memory
- vision::image::resize
- vision::image::resize_preserve_aspect_ratio
- vision::image::resize_preserve_aspect_ratio_hwc
- vision::image::save
- vision::imagenet::load_from_dir
- vision::imagenet::load_image
- vision::imagenet::load_image_and_resize
- vision::imagenet::load_image_and_resize224
- vision::imagenet::load_image_and_resize224_from_memory
- vision::imagenet::load_image_and_resize_from_memory
- vision::imagenet::load_image_from_memory
- vision::imagenet::normalize
- vision::imagenet::save_image
- vision::imagenet::top
- vision::imagenet::unnormalize
- vision::inception::v3
- vision::mnist::load_dir
- vision::mobilenet::v2
- vision::resnet::resnet101
- vision::resnet::resnet101_no_final_layer
- vision::resnet::resnet150_no_final_layer
- vision::resnet::resnet152
- vision::resnet::resnet18
- vision::resnet::resnet18_no_final_layer
- vision::resnet::resnet34
- vision::resnet::resnet34_no_final_layer
- vision::resnet::resnet50
- vision::resnet::resnet50_no_final_layer
- vision::squeezenet::v1_0
- vision::squeezenet::v1_1
- vision::vgg::vgg11
- vision::vgg::vgg11_bn
- vision::vgg::vgg13
- vision::vgg::vgg13_bn
- vision::vgg::vgg16
- vision::vgg::vgg16_bn
- vision::vgg::vgg19
- vision::vgg::vgg19_bn
- with_grad
Type Definitions
- Result
- nn::Conv1D
- nn::Conv2D
- nn::Conv3D
- nn::ConvConfig
- nn::ConvTranspose1D
- nn::ConvTranspose2D
- nn::ConvTranspose3D
- nn::ConvTransposeConfig