Package io.github.kirstenali.deepj.tensor

Record Class CausalAttentionResult

java.lang.Object

java.lang.Record

io.github.kirstenali.deepj.tensor.CausalAttentionResult

public record CausalAttentionResult(Tensor output, Tensor normalization, Tensor accumulatedOutput)
extends Record

Output, row statistics and unrounded FP32 output needed by tiled backward.

Constructor Summary

Constructors

Constructor	Description
CausalAttentionResult(Tensor output, Tensor normalization)
CausalAttentionResult(Tensor output, Tensor normalization, Tensor accumulatedOutput)	Creates an instance of a CausalAttentionResult record class.

Method Summary

Modifier and Type	Method	Description
Tensor	accumulatedOutput()	Returns the value of the accumulatedOutput record component.
final boolean	equals(Object o)	Indicates whether some other object is "equal to" this one.
final int	hashCode()	Returns a hash code value for this object.
Tensor	normalization()	Returns the value of the normalization record component.
Tensor	output()	Returns the value of the output record component.
final String	toString()	Returns a string representation of this record class.

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Constructor Details

CausalAttentionResult

public CausalAttentionResult(Tensor output,
 Tensor normalization)

CausalAttentionResult

public CausalAttentionResult(Tensor output,
 Tensor normalization,
 Tensor accumulatedOutput)

Creates an instance of a CausalAttentionResult record class.

Parameters:

output - the value for the output record component

normalization - the value for the normalization record component

accumulatedOutput - the value for the accumulatedOutput record component

Method Details

toString

public final String toString()

Returns a string representation of this record class. The representation contains the name of the class, followed by the name and value of each of the record components.

Specified by:

toString in class Record

Returns:

a string representation of this object

hashCode

public final int hashCode()

Returns a hash code value for this object. The value is derived from the hash code of each of the record components.

Specified by:

hashCode in class Record

Returns:

a hash code value for this object

equals

public final boolean equals(Object o)

Indicates whether some other object is "equal to" this one. The objects are equal if the other object is of the same class and if all the record components are equal. All components in this record class are compared with Objects::equals(Object,Object).

Specified by:

equals in class Record

Parameters:

o - the object with which to compare

Returns:

true if this object is the same as the o argument; false otherwise.

output

public Tensor output()

Returns the value of the output record component.

Returns:

the value of the output record component

normalization

public Tensor normalization()

Returns the value of the normalization record component.

Returns:

the value of the normalization record component

accumulatedOutput

public Tensor accumulatedOutput()

Returns the value of the accumulatedOutput record component.

Returns:

the value of the accumulatedOutput record component