autogradEngine/tensor library

Classes

Node
Tensor<T>

Functions

add(Tensor<Scalar> a, Tensor<Scalar> b) → Tensor<Scalar>
add3D(Tensor<Tensor3D> a, Tensor<Tensor3D> b) → Tensor<Tensor3D>
///////////////////////////////////////////////////////////////////////////// /////////////////////////////////////////////////////////////////////////////
addMatrix(Tensor<Matrix> a, Tensor<Matrix> b) → Tensor<Matrix>
///////////////////////////////////////////////////////////////////////////// /////////////////////////////////////////////////////////////////////////////
addMatrixAndVector(Tensor<Matrix> m, Tensor<Vector> v) → Tensor<Matrix>
addScalar(Tensor<Vector> v, double s) → Tensor<Vector>
addScalarToMatrix(Tensor<Matrix> m, Tensor<Scalar> s) → Tensor<Matrix>
addVector(Tensor<Vector> a, Tensor<Vector> b) → Tensor<Vector>
///////////////////////////////////////////////////////////////////////////// /////////////////////////////////////////////////////////////////////////////
binaryCrossEntropy(Tensor<Scalar> prediction, Tensor<Scalar> target) → Tensor<Scalar>
concatenate(Tensor<Vector> a, Tensor<Vector> b) → Tensor<Vector>
concatenate3D(Tensor<Tensor3D> a, Tensor<Tensor3D> b) → Tensor<Tensor3D>
concatenateMatricesByColumn(List<Tensor<Matrix>> matrices) → Tensor<Matrix>
conv2d(Tensor<Matrix> input, Tensor<Matrix> kernel, {String padding = 'valid'}) → Tensor<Matrix>
dot(Tensor<Vector> a, Tensor<Vector> b) → Tensor<Scalar>
elementWiseMultiply(Tensor<Vector> a, Tensor<Vector> b) → Tensor<Vector>
elementWiseMultiply3D(Tensor<Tensor3D> a, Tensor<Tensor3D> b) → Tensor<Tensor3D>
elementWiseMultiplyMatrix(Tensor<Matrix> a, Tensor<Matrix> b) → Tensor<Matrix>
matMul(Tensor<Matrix> a, Tensor<Matrix> b) → Tensor<Matrix>
Performs matrix multiplication, using the GPU for large matrices.
matVecMul(Tensor<Matrix> M, Tensor<Vector> v) → Tensor<Vector>
mse(Tensor<Vector> predictions, Tensor<Vector> targets) → Tensor<Scalar>
mseMatrix(Tensor<Matrix> predictions, Tensor<Matrix> targets) → Tensor<Scalar>
multiply(Tensor<Scalar> a, Tensor<Scalar> b) → Tensor<Scalar>
///////////////////////////////////////////////////////////////////////////// /////////////////////////////////////////////////////////////////////////////
multiplyScalar(Tensor<Scalar> a, Tensor<Scalar> b) → Tensor<Scalar>
padMatrix(Matrix input, int padding) → Matrix
///////////////////////////////////////////////////////////////////////////// /////////////////////////////////////////////////////////////////////////////
relu(Tensor<Vector> v) → Tensor<Vector>
reluMatrix(Tensor<Matrix> m) → Tensor<Matrix>
reshapeVectorToMatrix(Tensor<Vector> v, int numRows, int numCols) → Tensor<Matrix>
scaleMatrix(Tensor<Matrix> m, double s) → Tensor<Matrix>
selectRow(Tensor<Matrix> m, int rowIndex) → Tensor<Vector>
sigmoid(Tensor<Vector> v) → Tensor<Vector>
sigmoidMatrix(Tensor<Matrix> m) → Tensor<Matrix>
sigmoidScalar(Tensor<Scalar> s) → Tensor<Scalar>
softplus(Tensor<Vector> v) → Tensor<Vector>
///////////////////////////////////////////////////////////////////////////// /////////////////////////////////////////////////////////////////////////////
sum(Tensor<Vector> v) → Tensor<Scalar>
sumMatrix(Tensor<Matrix> m) → Tensor<Scalar>
tanhMatrix(Tensor<Matrix> m) → Tensor<Matrix>
transpose(Tensor<Matrix> a) → Tensor<Matrix>
vectorExp(Tensor<Vector> v) → Tensor<Vector>
vectorLog(Tensor<Vector> v) → Tensor<Vector>
vectorTanh(Tensor<Vector> v) → Tensor<Vector>