com.intel.analytics.bigdl.nn

abstractnn

package abstractnn

Visibility
  1. Public
  2. All

Type Members

  1. abstract class AbstractCriterion[A <: Activity, B <: Activity, T] extends Serializable

    AbstractCriterion is an abstract class the concrete criterion should extend.

  2. abstract class AbstractModule[A <: Activity, B <: Activity, T] extends Serializable with InferShape

    Module is the basic component of a neural network.

  3. trait Activity extends AnyRef

    Activity is a trait which represents the concept of neural input within neural networks.

  4. sealed trait DataFormat extends AnyRef

    DataFormat are used to specify the data format of the input and output data when data is 2-D images.

  5. class EmptyGradInput extends Activity with Serializable

    Sometimes a module may not have gradInput in the backward(e.

  6. trait IdentityOutputShape extends InferShape

  7. trait InferShape extends AnyRef

  8. trait Initializable extends AnyRef

    The trait that contains setInitMethod

  9. class InvalidLayer extends RuntimeException

  10. abstract class TensorCriterion[T] extends AbstractCriterion[Tensor[T], Tensor[T], T]

    TensorCriterion is an abstract sub-class of AbstractCriterion, whose input and output type both are Tensor.

  11. abstract class TensorModule[T] extends AbstractModule[Tensor[T], Tensor[T], T]

    TensorModule is an abstract sub-class of AbstractModule, whose input and output type both are Tensor.

Value Members

  1. object Activity

  2. object DataFormat

  3. object SizeAverageStatus extends Enumeration

Ungrouped