Package io.github.kirstenali.deepj.layers

Class TiedEmbeddingProjection

java.lang.Object

io.github.kirstenali.deepj.layers.TiedEmbeddingProjection

All Implemented Interfaces:

Layer, Projection, Trainable

public final class TiedEmbeddingProjection
extends Object
implements Projection

Predicts tokens using the same parameter that supplies the input embeddings.

Constructor Summary

Constructors

Constructor	Description
TiedEmbeddingProjection(Parameter embedding)

Method Summary

Modifier and Type	Method	Description
Tensor	forward(Tensor value)
List<Parameter>	parameters()

Methods inherited from class java.lang.Object

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

Methods inherited from interface io.github.kirstenali.deepj.layers.Layer

forward, supportsIncrementalInference

Methods inherited from interface io.github.kirstenali.deepj.training.Trainable

zeroGrad

Constructor Details

TiedEmbeddingProjection

public TiedEmbeddingProjection(Parameter embedding)

Method Details

forward

public Tensor forward(Tensor value)

Specified by:

forward in interface Layer

parameters

public List<Parameter> parameters()

Specified by:

parameters in interface Trainable