Package org.jetbrains.kotlinx.dl.api.core.layer.reshaping

Types

AbstractCropping

abstract class AbstractCropping(croppingInternal: Array<IntArray>, name: String) : Layer

Abstract Cropping layer used as the base layer for all the cropping layers.

AbstractUpSampling

abstract class AbstractUpSampling(sizeInternal: IntArray, interpolationInternal: InterpolationMethod, name: String) : Layer

Abstract UpSampling layer used as the base layer for all the upsampling layers.

AbstractZeroPadding

abstract class AbstractZeroPadding(name: String) : Layer

Abstract Zero Padding layer used as the base layer for all the ZeroPadding layers.

Cropping1D

class Cropping1D(cropping: IntArray, name: String) : AbstractCropping

Cropping layer for 1D data (e.g. audio, time-series)

Cropping2D

class Cropping2D(cropping: Array<IntArray>, name: String) : AbstractCropping

Cropping layer for 2D data (e.g. images)

Cropping3D

class Cropping3D(cropping: Array<IntArray>, name: String) : AbstractCropping

Cropping layer for 3D data (e.g. video, spatio-temporal)

Flatten

class Flatten(name: String) : Layer

Flattens the input. Does not affect the batch size.

InterpolationMethod

enum InterpolationMethod : Enum<InterpolationMethod>

Type of interpolation method.

Permute

class Permute(dims: IntArray, name: String) : Layer

Permutes the dimensions of the input according to a given pattern. Input shape: Arbitrary Output shape: Same as input shape, but with the dimensions re-ordered according to the specified pattern.

RepeatVector

class RepeatVector(n: Int, name: String) : Layer

Layer that repeats the input n times.

Reshape

class Reshape(targetShape: List<Int>, name: String) : Layer

Layer that reshapes inputs into the given shape.

UpSampling1D

class UpSampling1D(size: Int, name: String) : AbstractUpSampling

Upsampling layer for 1D input.

UpSampling2D

class UpSampling2D(size: IntArray, interpolation: InterpolationMethod, name: String) : AbstractUpSampling

Upsampling layer for 2D input.

UpSampling3D

class UpSampling3D(size: IntArray, name: String) : AbstractUpSampling

Upsampling layer for 3D input.

ZeroPadding1D

class ZeroPadding1D : AbstractZeroPadding

Zero-padding layer for 1D input (e.g. audio). This layer can add zeros in the rows of the audio tensor

ZeroPadding2D

class ZeroPadding2D : AbstractZeroPadding

Zero-padding layer for 2D input (e.g. picture). This layer can add rows and columns of zeros at the top, bottom, left and right side of an image tensor.

ZeroPadding3D

class ZeroPadding3D : AbstractZeroPadding

Zero-padding layer for 3D input (e.g. video). This layer can add zeros in the rows, cols and depth of a video tensor.