Package io.github.kirstenali.deepj.tensor

Class MatrixShapeChecks

java.lang.Object

io.github.kirstenali.deepj.tensor.MatrixShapeChecks

public final class MatrixShapeChecks
extends Object

Common validation for the existing rank-two CPU and native attention kernels.

Method Summary

Modifier and Type	Method	Description
static void	requireCausal(Tensor input, int sequenceLength)
static void	requireMatrix(Tensor input)
static void	requireRotary(Tensor input, Tensor cosine, Tensor sine, int sequenceLength)
static void	requireRowMaximum(Tensor input)

Methods inherited from class java.lang.Object

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

Method Details

requireMatrix

public static void requireMatrix(Tensor input)

requireRowMaximum

public static void requireRowMaximum(Tensor input)

requireCausal

public static void requireCausal(Tensor input,
 int sequenceLength)

requireRotary

public static void requireRotary(Tensor input,
 Tensor cosine,
 Tensor sine,
 int sequenceLength)