Module cast

burn0.16.0

Module cast

Module Items

Traits

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

Module cast

Settings

Help

Expand description

Tensor element casting.

Traits§

ToElement
A generic trait for converting a value to a number. Adapted from num_traits::ToPrimitive to support bool.