R/lenght.R

In dfalbel/capsnet: Dynamic Routing Between Capsules implementation in Keras

layer_length <- function(object, name = NULL, trainable = TRUE) {
  keras::create_layer(Length, object, list(name = name, trainable = trainable))
}

Length <- R6::R6Class(
  "Length",

  inherit = keras::KerasLayer,

  public = list(

    call = function(inputs, mask = NULL) {
      keras::k_sqrt(keras::k_sum(keras::k_square(inputs), -1L))
    },

    compute_output_shape = function(input_shape) {
      reticulate::tuple(input_shape[1:(length(input_shape) - 1)])
    }

  )
)