Package io.github.kirstenali.deepj.layers

Class ProjectionOps

java.lang.Object

io.github.kirstenali.deepj.layers.ProjectionOps

public final class ProjectionOps
extends Object

Independent leading positions share one projection; public activation axes remain intact.

Method Summary

Modifier and Type	Method	Description
static Tensor	affine(Tensor input, Tensor weight, Tensor bias)
static Tensor	linear(Tensor input, Tensor weight)
static Tensor	tied(Tensor input, Tensor embedding)

Methods inherited from class java.lang.Object

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

Method Details

linear

public static Tensor linear(Tensor input,
 Tensor weight)

affine

public static Tensor affine(Tensor input,
 Tensor weight,
 Tensor bias)

tied

public static Tensor tied(Tensor input,
 Tensor embedding)