Expand description
The struct definitions for all TensorXD, Tensor trait, and more.
Structs
A fake tensor that holds a UniqueId and a type T that is HasArrayType.
This is created and stored in GradientTape operations to access gradient data
for a tensor that the GradientTape doesn’t have ownership of.
A 0d [Tensor] with shape (). Backed by data f32.
A 1d [Tensor] with shape (M, ). Backed by data [f32; M].
A 2d [Tensor] with shape (M, N). Backed by data [[f32; N]; M].
A 3d [Tensor] with shape (M, N, O). Backed by data [[[f32; O]; N]; M].
A 4d [Tensor] with shape (M, N, O, P). Backed by data [[[[f32; P]; O]; N]; M].
Traits
Something that has HasArrayType, and also can return a reference to or mutate Self::Array.
Something that can be turned into a PhantomTensor
Something that has parameters that can be randomized from a generic distribution.
The main tensor trait. A tensor consists of mainly 1. an array, 2. a device, 3. a unique id.
Something that can be created - currently only implemented for tensors with no tapes.
Functions
Runs backprop algorithm with all operations contained in the tape that t has.