R/differential-rep-t-aperm.r

In salad: Simple Automatic Differentiation

# rep
rep.differential <- function(x, ...) {
  x <- unclass(x)
  L <- vector("list", length(x))
  for(i in seq_along(x)) L[[i]] <- rep(x[[i]], ...)
  names(L) <- names(x)
  class(L) <- "differential"
  L
}


# transposition
t.differential <- function(x) {
  x <- unclass(x)
  L <- vector("list", length(x))
  for(i in seq_along(x)) L[[i]] <- t(x[[i]])
  names(L) <- names(x)
  class(L) <- "differential"
  L
}

# aperm
aperm.differential <- function(a, perm = NULL, resize = TRUE, ...) {
  a <- unclass(a)
  x <- vector("list", length(a))
  for(i in seq_along(a)) x[[i]] <- aperm(a[[i]], perm, resize, ...)
  names(x) <- names(a)
  class(x) <- "differential"
  x
}