List of all items

Structs

• Bool
• Data
• DataSerialize
• DimIter
• DistributionSampler
• Float
• Int
• PrintOptions
• Shape
• T
• Tensor
• bf16
• container::TensorContainer
• f16
• ops::Conv1dBackward
• ops::Conv2dBackward
• ops::ConvOptions
• ops::ConvTransposeOptions
• ops::MaxPool1dBackward
• ops::MaxPool1dWithIndices
• ops::MaxPool2dBackward
• ops::MaxPool2dWithIndices
• ops::UnfoldOptions

Enums

• Distribution
• DistributionSamplerKind
• Precision
• Reader

Traits

• BasicAutodiffOps
• BasicOps
• Element
• ElementConversion
• ElementPrecision
• ElementRandom
• Numeric
• ReshapeArgs
• TensorKind
• backend::AutodiffBackend
• backend::Backend
• ops::ActivationOps
• ops::BoolTensorOps
• ops::IntTensorOps
• ops::ModuleOps
• ops::TensorOps

Macros

• check
• make_element

Functions

• activation::gelu
• activation::log_sigmoid
• activation::log_softmax
• activation::quiet_softmax
• activation::relu
• activation::sigmoid
• activation::silu
• activation::softmax
• activation::tanh
• loss::cross_entropy_with_logits
• module::adaptive_avg_pool1d
• module::adaptive_avg_pool2d
• module::avg_pool1d
• module::avg_pool2d
• module::conv1d
• module::conv2d
• module::conv_transpose1d
• module::conv_transpose2d
• module::embedding
• module::max_pool1d
• module::max_pool1d_with_indices
• module::max_pool2d
• module::max_pool2d_with_indices
• module::unfold4d
• ops::conv::calculate_conv_output_size
• ops::conv::calculate_conv_padding
• ops::conv::calculate_conv_transpose_output_size
• ops::conv::calculate_pool_output_size
• set_print_options

Type Aliases

• Device
• ops::BoolTensor
• ops::Device
• ops::FloatElem
• ops::FloatTensor
• ops::FullPrecisionBackend
• ops::IntElem
• ops::IntTensor