com.yahoo.tensor.functions

Class Softmax

java.lang.Object

com.yahoo.tensor.functions.TensorFunction

com.yahoo.tensor.functions.CompositeTensorFunction

com.yahoo.tensor.functions.Softmax

@Beta
public class Softmax
extends CompositeTensorFunction

Author:

bratseth

Field Summary

Fields

Modifier and Type	Field and Description
private TensorFunction	argument
private String	dimension

Constructor Summary

Constructors

Constructor and Description
Softmax(TensorFunction argument, String dimension)

Method Summary

Modifier and Type	Method and Description
List<TensorFunction>	functionArguments() Returns the function arguments of this node in the order they are applied
TensorFunction	replaceArguments(List<TensorFunction> arguments) Returns a copy of this tensor function with the arguments replaced by the given list of arguments.
PrimitiveTensorFunction	toPrimitive() Translate this function - and all of its arguments recursively - to a tree of primitive functions only.
String	toString(ToStringContext context) Return a string representation of this context.

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

evaluate

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

Field Detail

argument

private final TensorFunction argument

dimension

private final String dimension

Constructor Detail

Softmax

public Softmax(TensorFunction argument,
               String dimension)

Method Detail

functionArguments

public List<TensorFunction> functionArguments()

Description copied from class: TensorFunction

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

Specified by:

functionArguments in class TensorFunction

replaceArguments

public TensorFunction replaceArguments(List<TensorFunction> 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:

replaceArguments in class TensorFunction

toPrimitive

public PrimitiveTensorFunction 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

Returns:

a tree of primitive functions implementing this

toString

public String toString(ToStringContext context)

Description copied from class: TensorFunction

Return a string representation of this context.

Specified by:

toString in class TensorFunction

Parameters:

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