Package io.github.kirstenali.deepj.tensor

Class HeadLayout

java.lang.Object

io.github.kirstenali.deepj.tensor.HeadLayout

public final class HeadLayout
extends Object

Batch and head axes are distinct; packed matrices are temporary native-kernel views only.

Method Summary

Modifier and Type	Method	Description
static Tensor	merge(Tensor input, int heads)
static Tensor	pack(Tensor input)
static void	require(Tensor input, int heads)
static Tensor	split(Tensor input, int heads)
static int	totalHeads(Tensor input, int heads)

Methods inherited from class java.lang.Object

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

Method Details

split

public static Tensor split(Tensor input,
 int heads)

merge

public static Tensor merge(Tensor input,
 int heads)

pack

public static Tensor pack(Tensor input)

require

public static void require(Tensor input,
 int heads)

totalHeads

public static int totalHeads(Tensor input,
 int heads)