Module loss
burn
0.16.0
Module loss
Module Items
Functions
In burn::
tensor
Modules
activation
backend
cast
container
loss
module
ops
quantization
repr
Macros
dequant_op_quant
make_element
Structs
Bool
Bytes
Data
DataSerialize
DimIter
DistributionSampler
Float
Int
PrintOptions
Shape
T
Tensor
TensorData
Transaction
bf16
f16
Enums
DType
DataError
Distribution
DistributionSamplerKind
FloatDType
Precision
TensorPrimitive
Traits
BasicAutodiffOps
BasicOps
BroadcastArgs
Element
ElementComparison
ElementConversion
ElementPrecision
ElementRandom
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.