R/types.R

In TobiasHerr/tensorflow-r-fork: R Interface to 'TensorFlow'

Defines functions shape

Documented in shape

#' Tensor shape
#'
#' @param ... Tensor dimensions
#'
#' @export
shape <- function(...) {
  values <- list(...)
  lapply(values, function(value) {
    if (!is.null(value))
      as.integer(value)
    else
      NULL
  })
}