Package com.yahoo.tensor.functions

Class Softmax<NAMETYPE extends Name>

java.lang.Object

com.yahoo.tensor.functions.TensorFunction<NAMETYPE>

com.yahoo.tensor.functions.CompositeTensorFunction<NAMETYPE>

com.yahoo.tensor.functions.Softmax<NAMETYPE>

public class Softmax<NAMETYPE extends Name>
extends CompositeTensorFunction<NAMETYPE>

Author:

bratseth

Constructor Summary

Constructors

Constructor	Description
Softmax(TensorFunction<NAMETYPE> argument, java.lang.String dimension)

Method Summary

Modifier and Type	Method	Description
java.util.List<TensorFunction<NAMETYPE>>	arguments()	Returns the function arguments of this node in the order they are applied
static TensorType	outputType(TensorType inputType, java.lang.String dimension)
PrimitiveTensorFunction<NAMETYPE>	toPrimitive()	Translate this function - and all of its arguments recursively - to a tree of primitive functions only.
java.lang.String	toString(ToStringContext context)	Return a string representation of this context.
TensorFunction<NAMETYPE>	withArguments(java.util.List<TensorFunction<NAMETYPE>> arguments)	Returns a copy of this tensor function with the arguments replaced by the given list of arguments.

Methods inherited from class com.yahoo.tensor.functions.CompositeTensorFunction

evaluate, type

Methods inherited from class com.yahoo.tensor.functions.TensorFunction

evaluate, toString

Methods inherited from class java.lang.Object

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

Constructor Detail

Softmax

public Softmax(TensorFunction<NAMETYPE> argument,
               java.lang.String dimension)

Method Detail

outputType

public static TensorType outputType(TensorType inputType,
                                    java.lang.String dimension)

arguments

public java.util.List<TensorFunction<NAMETYPE>> arguments()

Description copied from class: TensorFunction

Returns the function arguments of this node in the order they are applied

Specified by:

arguments in class TensorFunction<NAMETYPE extends Name>

withArguments

public TensorFunction<NAMETYPE> withArguments(java.util.List<TensorFunction<NAMETYPE>> arguments)

Description copied from class: TensorFunction

Returns a copy of this tensor function with the arguments replaced by the given list of arguments.

Specified by:

withArguments in class TensorFunction<NAMETYPE extends Name>

toPrimitive

public PrimitiveTensorFunction<NAMETYPE> toPrimitive()

Description copied from class: TensorFunction

Translate this function - and all of its arguments recursively - to a tree of primitive functions only.

Specified by:

toPrimitive in class TensorFunction<NAMETYPE extends Name>

Returns:

a tree of primitive functions implementing this

toString

public java.lang.String toString(ToStringContext context)

Description copied from class: TensorFunction

Return a string representation of this context.

Specified by:

toString in class TensorFunction<NAMETYPE extends Name>

Parameters:

context - a context which must be passed to all nested functions when requesting the string value