Reshape

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

Layer that reshapes inputs into the given shape.

Input shape: (batch_size,) + any shape.

Output shape: (batch_size,) + target_shape.

0.2

Constructors

Reshape

fun Reshape(targetShape: List<Int>, name: String = "")

Creates Reshape object.

build

open override fun build(tf: Ops, input: Operand<Float>, isTraining: Operand<Boolean>, numberOfLosses: Operand<Float>?): Operand<Float>

open fun build(tf: Ops, input: List<Operand<Float>>, isTraining: Operand<Boolean>, numberOfLosses: Operand<Float>?): Operand<Float>

Extend this function to define variables in the layer and compute layer output.

invoke

operator fun invoke(vararg layers: Layer): Layer

Important part of functional API. It takes layers as input and saves them to the inboundLayers of the given layer.

toString

open override fun toString(): String

hasActivation

Returns True, if layer has internal activation function.

inboundLayers

Returns inbound layers.

name

Layer name. A new name is generated during model compilation when provided name is empty.

outboundLayers

Returns outbound layers.

outputShape

Output data tensor shape.

parentModel

Model where this layer is used.

targetShape

Target shape. List of integers, does not include the samples dimension (batch size)