List of all items

Structs

• BoolTensor
• Data
• DataSerialize
• DistributionSampler
• Shape
• Tensor
• container::TensorContainer
• f16
• ops::Conv1dBackward
• ops::Conv2dBackward
• ops::MaxPool2dBackward
• ops::MaxPool2dWithIndexes

Enums

• Distribution
• DistributionSamplerKind
• Precision

Traits

• Element
• ElementConversion
• ElementPrecision
• ElementRandom
• ElementValue
• backend::ADBackend
• backend::Backend
• ops::ModuleOps
• ops::Ones
• ops::TensorOps
• ops::Zeros

Macros

• make_element

Functions

• activation::gelu
• activation::log_softmax
• activation::relu
• activation::softmax
• loss::cross_entropy_with_logits
• module::conv1d
• module::conv2d
• module::embedding
• module::max_pool2d
• module::max_pool2d_with_indexes
• ops::conv::calculate_output_size
• ops::conv::calculate_padding