Skip to main content

Crate lumen_core

Crate lumen_core 

Source

Modules§

display
Pretty printing of tensors
utils

Macros§

bail
inplace_warning_doc
no_grad
s

Structs§

AutogradInfo
ContiguousStorageIndices
DimCoordinates
DimCoordinates
DimNCoordinates
FloatCategory
GradStore
IntCategory
Layout
NoAutograd
NoGradGuard
Shape
Slice
Storage
StorageArc
Tensor
TensorId
TensorIter
UncontiguousStorageIndices
Var
bf16
A 16-bit floating point type implementing the bfloat16 format.

Enums§

BinaryOp
CmpOp
D
DType
DynTensor
Error
FloatTensor
Indexer
IntTensor
Op
ReduceOp
StorageIndices
StorageIndices
TensorOrScalar
UnaryOp

Traits§

AutogradMetaT
DTypeConvert
Dim
Dims
FloatDType
IndexOp
IntDType
NumCategory
NumDType
ResettableIterator
ToTensor
WithDType

Functions§

dims0
dims1
dims2
dims3
dims4
dims5
is_grad_enabled
set_grad_enabled

Type Aliases§

Result