Module loss
burn
0.17.0
Module loss
Module Items
Functions
In burn::
tensor
Modules
activation
backend
cast
container
loss
module
ops
quantization
Macros
dequant_op_quant
make_element
s
Structs
Bool
Bytes
DimIter
DistributionSampler
Float
Int
PrintOptions
Shape
Slice
T
Tensor
TensorData
Tolerance
Transaction
bf16
f16
flex32
Enums
DType
DataError
Distribution
DistributionSamplerKind
FloatDType
Precision
TensorPrimitive
Constants
DEFAULT_ATOL
DEFAULT_RTOL
Traits
BasicAutodiffOps
BasicOps
BroadcastArgs
Element
ElementComparison
ElementConversion
ElementLimits
ElementPrecision
ElementRandom
IndexConversion
MovedimArgs
Numeric
RangesArg
ReshapeArgs
TensorKind
TensorMetadata
Functions
argsort
argwhere_data
cartesian_grid
check_closeness
chunk
narrow
read_sync
set_print_options
sort
sort_with_indices
split
split_with_sizes
try_read_sync
Type Aliases
Device
burn
::
tensor
Module
loss
Copy item path
Settings
Help
Summary
Expand description
The loss module.
Functions
§
cross_
entropy_
with_
logits
Computes the log softmax cross entropy between logits and target probabilities.