ggmlR (version 0.6.1)

ggml_layer_conv_2d: Create a Conv2D Layer Object

Description

Create a Conv2D Layer Object

Add 2D Convolution Layer

Usage

ggml_layer_conv_2d(
  model,
  filters,
  kernel_size,
  activation = NULL,
  input_shape = NULL,
  strides = c(1L, 1L),
  padding = "valid",
  name = NULL,
  trainable = TRUE
)
ggml_layer_conv_2d(
  model,
  filters,
  kernel_size,
  activation = NULL,
  input_shape = NULL,
  strides = c(1L, 1L),
  padding = "valid",
  name = NULL,
  trainable = TRUE
)

Value

A ggml_layer object.

The model object with the conv_2d layer appended (invisibly).

Arguments

model

A ggml_sequential_model object

filters

Number of output filters

kernel_size

Integer or vector of 2 integers for kernel height and width

activation

Activation function name: "relu", "sigmoid", "tanh", "softmax", or NULL

input_shape

Input shape c(H, W, C) - required for first layer only

strides

Integer or vector of 2 integers for stride

padding

"valid" (no padding) or "same" (preserve spatial dims)

name

Optional character name for the layer.

trainable

Logical; whether the layer weights are updated during training.

Examples

# \donttest{
model <- ggml_model_sequential() |>
  ggml_layer_conv_2d(32, c(3,3), activation = "relu",
                     input_shape = c(28, 28, 1))
# }