Package io.github.kirstenali.deepj.tensor

Class PackedLayouts

java.lang.Object

io.github.kirstenali.deepj.tensor.PackedLayouts

public final class PackedLayouts
extends Object

Explicit compatibility for packed matrix APIs,delegating their work to the same ND operators.

Method Summary

Modifier and Type	Method	Description
static Tensor	matmul(TensorBackend backend, Tensor left, Tensor right, int batches, boolean transposeLeft, boolean transposeRight)
static Tensor	merge(TensorBackend backend, Tensor input, int heads)
static Tensor	split(TensorBackend backend, Tensor input, int heads)

Methods inherited from class java.lang.Object

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

Method Details

matmul

public static Tensor matmul(TensorBackend backend,
 Tensor left,
 Tensor right,
 int batches,
 boolean transposeLeft,
 boolean transposeRight)

split

public static Tensor split(TensorBackend backend,
 Tensor input,
 int heads)

merge

public static Tensor merge(TensorBackend backend,
 Tensor input,
 int heads)