Module ops

Module ops 

Source
Expand description

composable operators for tensor manipulations and transformations, neural networks, and more

Traitsยง

Apply
Apply is a composable binary operator generally used to apply some object or function onto the caller to produce some output.
ApplyMut
ApplyMut provides an interface for mutable containers that can apply a function onto their elements, modifying them in place.
ApplyOnce
The ApplyOnce trait consumes the container and applies the given function to every element before returning a new container with the results.
ArrayLike
Decrement
Decrement is a chainable trait that defines a decrement method, effectively removing a single unit from the original object to create another
DecrementAxis
The DecrementAxis is used as a unary operator for removing a single axis from a multidimensional array or tensor-like structure.
DecrementMut
The DecrementMut trait defines a decrement method that operates in place, modifying the original object.
DefaultLike
FillLike
Increment
The Increment
IncrementAxis
The IncrementAxis trait defines a method enabling an axis to increment itself, effectively adding a new axis to the array.
IncrementMut
IsSquare
IsSquare is a trait for checking if the layout, or dimensionality, of a tensor is square.
MapInto
MapInto defines an interface for containers that can consume themselves to apply a given function onto each of their elements.
MapTo
MapTo establishes an interface for containers capable of applying a given function onto each of their elements, by reference.
MaskFill
This trait is used to fill an array with a value based on a mask. The mask is a boolean array of the same shape as the array.
NdLike
OnesLike
Unsqueeze
The Unsqueeze trait establishes an interface for a routine that unsqueezes an array, by inserting a new axis at a specified position. This is useful for reshaping arrays to meet specific dimensional requirements.
ZerosLike