Package

com.intel.analytics.bigdl.nn

onnx

Permalink

package onnx

Visibility
  1. Public
  2. All

Type Members

  1. class Reshape[T] extends AbstractModule[Activity, Tensor[T], T]

    Permalink

    Reshape the input tensor similar to numpy.reshape.

    Reshape the input tensor similar to numpy.reshape. First input is the data tensor, second input is a shape tensor which specifies the output shape. It outputs the reshaped tensor.

    T

    The numeric type in this module parameters.

Value Members

  1. object Gemm extends Serializable

    Permalink

  2. object Reshape extends Serializable

    Permalink

  3. object Shape extends Serializable

    Permalink

Ungrouped