Package io.github.kirstenali.deepj.autograd

Class ViewOps

java.lang.Object

io.github.kirstenali.deepj.autograd.ViewOps

public final class ViewOps
extends Object

Method Summary

Modifier and Type	Method	Description
static Tensor	cast(Tensor input, DType dtype)
static Tensor	contiguous(Tensor input)
static Tensor	copy(Tensor input)
static Tensor	expand(Tensor input, int... dimensions)
static Tensor	permute(Tensor input, int... axes)
static Tensor	reshape(Tensor input, int... dimensions)
static Tensor	select(Tensor input, int dimension, int index)
static Tensor	slice(Tensor input, int dimension, int start, int stop, int step)
static Tensor	transpose(Tensor input, int first, int second)
static Tensor	unsqueeze(Tensor input, int axis)
static Tensor	view(Tensor input, int... dimensions)

Methods inherited from class java.lang.Object

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

Method Details

unsqueeze

public static Tensor unsqueeze(Tensor input,
 int axis)

view

public static Tensor view(Tensor input,
 int... dimensions)

reshape

public static Tensor reshape(Tensor input,
 int... dimensions)

permute

public static Tensor permute(Tensor input,
 int... axes)

transpose

public static Tensor transpose(Tensor input,
 int first,
 int second)

expand

public static Tensor expand(Tensor input,
 int... dimensions)

select

public static Tensor select(Tensor input,
 int dimension,
 int index)

slice

public static Tensor slice(Tensor input,
 int dimension,
 int start,
 int stop,
 int step)

contiguous

public static Tensor contiguous(Tensor input)

copy

public static Tensor copy(Tensor input)

cast

public static Tensor cast(Tensor input,
 DType dtype)