GlobalAvgPool2D

class GlobalAvgPool2D(name: String) : Layer

Global average pooling operation for 2D data (images and so on).

NOTE: Works with tensors which must have rank 4 (batch, height, width, channels).

Input shape: 4D tensor with shape (batch_size, rows, cols, channels).

Output shape: 2D tensor with shape (batch_size, channels).

Since

0.2

Constructors

GlobalAvgPool2D

fun GlobalAvgPool2D(name: String = "")

Creates GlobalAvgPool2D object.

Functions

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

Properties

hasActivation

open override val hasActivation: Boolean

Returns True, if layer has internal activation function.

inboundLayers

var inboundLayers: MutableList<Layer>

Returns inbound layers.

name

var name: String

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

outboundLayers

var outboundLayers: MutableList<Layer>

Returns outbound layers.

outputShape

lateinit var outputShape: TensorShape

Output data tensor shape.

parentModel

var parentModel: GraphTrainableModel? = null

Model where this layer is used.

GlobalAvgPool2D

Since

Constructors

On this page

On this page

Functions

On this page

On this page

Properties

On this page

On this page