Docs.rs
  • burn-tensor-0.3.0
    • burn-tensor 0.3.0
    • Docs.rs crate page
    • MIT/Apache-2.0
    • Links
    • Repository
    • crates.io
    • Source
    • Owners
    • nathanielsimard
    • Dependencies
      • burn-tensor-testgen ^0.3.0 normal
      • derive-new ^0.5 normal
      • half ^1.6 normal
      • nanoid ^0.4 normal
      • num-traits ^0.2 normal
      • rand ^0.8 normal
      • serde ^1.0 normal
    • Versions
    • 0% of the crate is documented
  • Go to latest version
  • Platform
    • i686-pc-windows-msvc
    • i686-unknown-linux-gnu
    • x86_64-apple-darwin
    • x86_64-pc-windows-msvc
    • x86_64-unknown-linux-gnu
  • Feature flags
  • docs.rs
    • About docs.rs
    • Badges
    • Builds
    • Metadata
    • Shorthand URLs
    • Download
    • Rustdoc JSON
    • Build queue
    • Privacy policy
  • Rust
    • Rust website
    • The Book
    • Standard Library API Reference
    • Rust by Example
    • The Cargo Guide
    • Clippy Documentation
logo

logo

Crate burn_tensor

  • Macros
  • Structs
  • Enums
  • Traits
  • Functions
?
Change settings

List of all items

Structs

  • BoolTensor
  • Data
  • DataSerialize
  • DistributionSampler
  • Shape
  • Tensor
  • f16

Enums

  • Distribution
  • DistributionSamplerKind
  • Precision

Traits

  • Element
  • ElementConversion
  • ElementPrecision
  • ElementRandom
  • ElementValue
  • backend::ADBackend
  • backend::Backend
  • backend::Gradients
  • 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::embedding