Package org.apache.wayang.basic.model.op.nn

Class CrossEntropyLoss

java.lang.Object

org.apache.wayang.basic.model.op.Op

org.apache.wayang.basic.model.op.nn.CrossEntropyLoss

All Implemented Interfaces:

java.io.Serializable

public class CrossEntropyLoss
extends Op

See Also:

Serialized Form

Nested Class Summary

Nested classes/interfaces inherited from class org.apache.wayang.basic.model.op.Op

Op.DType

Field Summary

Fields inherited from class org.apache.wayang.basic.model.op.Op

dType, fromList, name

Constructor Summary

Constructors

Constructor	Description
CrossEntropyLoss(int labels)
CrossEntropyLoss(int labels, java.lang.String name)

Method Summary

Modifier and Type	Method	Description
Op.DType	getDType()
int	getLabels()
int	inputsRequired()

Methods inherited from class org.apache.wayang.basic.model.op.Op

getFromList, getName, with

Methods inherited from class java.lang.Object

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

Constructor Detail

CrossEntropyLoss

public CrossEntropyLoss(int labels)

CrossEntropyLoss

public CrossEntropyLoss(int labels,
                        java.lang.String name)

Method Detail

getLabels

public int getLabels()

getDType

public Op.DType getDType()

Overrides:

getDType in class Op

inputsRequired

public int inputsRequired()

Specified by:

inputsRequired in class Op