Package io.github.kirstenali.deepj.tensor

Class TensorManipulations

java.lang.Object

io.github.kirstenali.deepj.tensor.TensorStorageAccess

io.github.kirstenali.deepj.tensor.TensorManipulations

Direct Known Subclasses:

Tensor

public abstract class TensorManipulations
extends TensorStorageAccess

Shape transformations retain one Tensor system and participate in first-order autograd.

Method Summary

Modifier and Type	Method	Description
final Tensor	contiguous()
final Tensor	copyFrom(Tensor source)	Independent value assignment with an explicit cast to this tensor's dtype; shapes must match.
final Tensor	expand(int... shape)
final Tensor	indexAddInPlace(int axis, int[] indices, Tensor updates)
final Tensor	indexSelect(int axis, int... indices)
final Tensor	maxAxes(int... axes)
final Tensor	maximum(int axis, boolean keep)	Tied maxima share their gradient equally.
final Tensor	mean(int axis, boolean keep)
final Tensor	meanAxes(int... axes)
static Tensor	ones(DType dtype, int... shape)
final Tensor	permute(int... axes)
static Tensor	random(DType dtype, Random random, int... shape)
boolean	requiresGrad()
final Tensor	reshape(int... shape)
final Tensor	select(int axis, int index)
final Tensor	slice(int axis, int start, int stop)
final Tensor	slice(int axis, int start, int stop, int step)
final Tensor	softmax(int axis)
final Tensor	sum(int axis, boolean keepDimensions)
final Tensor	sumAxes(int... axes)
protected abstract Tensor	tensor()
final Tensor	to(DType dtype)
final Tensor	transpose(int first, int second)
final Tensor	unsqueeze(int axis)
final Tensor	variance(int axis, boolean keep)	Population variance; use the correction overload for a different denominator.
final Tensor	variance(int axis, boolean keep, int correction)
final Tensor	varianceAxes(int[] axes, boolean keep, int correction)
final Tensor	view(int... shape)
static Tensor	zeros(DType dtype, int... shape)

Methods inherited from class io.github.kirstenali.deepj.tensor.TensorStorageAccess

coversStorage, dtype, get, get, getGpuTag, gradientState, hasHostStorage, hasSameShape, hasShape, isContiguous, materialize, matrixCols, matrixRows, numel, rank, rawData, releaseDeviceBufferRetention, retainsDeviceBuffer, set, setAt, setGpuTag, shape, size, storageBytes, storageOffset, strides, toFloatArray

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Details

zeros

public static Tensor zeros(DType dtype,
 int... shape)

ones

public static Tensor ones(DType dtype,
 int... shape)

random

public static Tensor random(DType dtype,
 Random random,
 int... shape)

tensor

protected abstract Tensor tensor()

requiresGrad

public boolean requiresGrad()

view

public final Tensor view(int... shape)

unsqueeze

public final Tensor unsqueeze(int axis)

reshape

public final Tensor reshape(int... shape)

permute

public final Tensor permute(int... axes)

transpose

public final Tensor transpose(int first,
 int second)

select

public final Tensor select(int axis,
 int index)

expand

public final Tensor expand(int... shape)

contiguous

public final Tensor contiguous()

to

public final Tensor to(DType dtype)

copyFrom

public final Tensor copyFrom(Tensor source)

Independent value assignment with an explicit cast to this tensor's dtype; shapes must match.

slice

public final Tensor slice(int axis,
 int start,
 int stop,
 int step)

slice

public final Tensor slice(int axis,
 int start,
 int stop)

sum

public final Tensor sum(int axis,
 boolean keepDimensions)

sumAxes

public final Tensor sumAxes(int... axes)

mean

public final Tensor mean(int axis,
 boolean keep)

meanAxes

public final Tensor meanAxes(int... axes)

variance

public final Tensor variance(int axis,
 boolean keep,
 int correction)

variance

public final Tensor variance(int axis,
 boolean keep)

Population variance; use the correction overload for a different denominator.

varianceAxes

public final Tensor varianceAxes(int[] axes,
 boolean keep,
 int correction)

maximum

public final Tensor maximum(int axis,
 boolean keep)

Tied maxima share their gradient equally.

maxAxes

public final Tensor maxAxes(int... axes)

softmax

public final Tensor softmax(int axis)

indexSelect

public final Tensor indexSelect(int axis,
 int... indices)

indexAddInPlace

public final Tensor indexAddInPlace(int axis,
 int[] indices,
 Tensor updates)