burn-core-0.17.1
- burn-core 0.17.1
- Docs.rs crate page
- MIT OR Apache-2.0
- Links
- Repository
- crates.io
- Source
- Owners
- nathanielsimard
- Dependencies
- ahash ^0.8.11 normal
  bincode ^2.0.1 normal
  burn-common ^0.17.1 normal
  burn-cuda ^0.17.1 normal optional
  burn-dataset ^0.17.1 normal optional
  burn-derive ^0.17.1 normal
  burn-remote ^0.17.1 normal optional
  burn-rocm ^0.17.1 normal optional
  burn-router ^0.17.1 normal optional
  burn-tch ^0.17.1 normal optional
  burn-tensor ^0.17.1 normal
  burn-wgpu ^0.17.1 normal optional
  data-encoding ^2.9.0 normal
  derive-new ^0.7.0 normal
  flate2 ^1.1.0 normal optional
  half ^2.6.0 normal
  hashbrown ^0.15.2 normal
  log ^0.4.27 normal optional
  num-traits ^0.2.19 normal
  rand ^0.9.1 normal
  regex ^1.11.1 normal optional
  rmp-serde ^1.3.0 normal optional
  serde ^1.0.218 normal
  serde_json ^1.0.140 normal
  spin ^0.10.0 normal
  thiserror ^2.0.11 normal optional
  uuid ^1.16.0 normal
  burn-autodiff ^0.17.1 dev
  burn-dataset ^0.17.1 dev
  burn-ndarray ^0.17.1 dev
  portable-atomic-util ^0.2.4 normal
- Versions
- 100% of the crate is documented
Go to latest version
Platform
- i686-unknown-linux-gnu
- x86_64-unknown-linux-gnu
Feature flags

docs.rs

Rust

burn_core0.17.1

Module loss

Module Items

Functions

In burn_core::tensor

burn_core::tensor

Module loss

Expand description

The loss module.

Functions§

cross_entropy_with_logits: Computes the log softmax cross entropy between logits and target probabilities.