List of all items

Structs

• Device
• NoGradTrack
• optimizers::Adam
• optimizers::AdamConfig
• serialization::StructDeserializer
• serialization::StructSerializer
• tensor::Shape
• tensor::Tensor

Enums

• DeviceType
• serialization::FieldValue
• serialization::Format
• serialization::SerializationError

Traits

• optimizers::Optimizer
• serialization::FromFieldValue
• serialization::Serializable
• serialization::StructSerializable
• serialization::ToFieldValue

Functions

• clear_gradients
• cuda_device_count
• cuda_is_available
• current_device
• get_default_device
• is_grad_enabled
• set_default_device
• set_grad_enabled
• with_device
• with_no_grad

Type Aliases

• serialization::SerializationResult