DTensor
Interface for a differentiable tensor.
DTensor is an interface that is implemented as either
a FloatTensor, which is a wrapper around an array of floats, or
a ForwardTensor for forward differentiation, or
a ReverseTensor for reverse mode differentiation.
Functions
Properties
Inheritors
Extensions
Returns the primal value as a FloatTensor.
Takes a tensor of shape T, a shape B, and a tensor of shape T and returns a tensor of shape T which is the inner product of the two.
Given the shape A,B,D (where A, B and D are lists of Ints, and D is possibly empty), this function converts a tensor of shape A,B,D to a tensor of shape B,A,D, shuffling the data so that the element at position i,j,k is at position j,i,k (where i, j and k are lists of integers corresponding to the shape A, B, and D).
Returns the current (this) tensor updated by the new tensor (new) scaled by momentum
Given the shape A,B,D (where A, B and D are lists of Ints, and D is possibly empty), this function converts a tensor of shape D,A,B to a tensor of shape D,B,A, shuffling the data so that the element at position i,j,k is at position i,k,j (where i, j and k are lists of integers corresponding to the shape D, B, and A).
Sum over given axes. If keepDims is true, the original rank of input is preserved. Otherwise, the dimensions provided by axis are removed from the output shape.