max/ourMELONS
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions

Optimized squeeze()

Browse source
This commit is contained in:
Waldir Leoncio 2020-07-14 11:18:01 +02:00
parent bb44d0cfd4
commit a27f303f4a
2 changed files with 7 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
R/squeeze.R
View file

@ -6,17 +6,10 @@
#' a 3-by-1-by-1-by-2 array, then squeeze(A) returns a 3-by-2 matrix. If A is a
#' row vector, column vector, scalar, or an array with no dimensions of length
#' 1, then squeeze returns the input A.
#' @note This is basically a wrapper of drop() with a minor adjustment to adapt
#' the output to what happens on Matlab
#' @param A input or array matrix
#' @return An array with the same elements as the input array, but with
#' dimensions of length 1 removed.
#' @author Waldir Leoncio
squeeze <- function(A) {
A <- as.array(A)
dim_1 <- which(dim(A) == 1)
B <- array(A, dim = dim(A)[-dim_1])
# Workaround to match Matlab behavior
if (length(dim(B)) == 1) B <- as.matrix(B)
return(B)
}
squeeze <- function(A) as.matrix(drop(A))

4
man/squeeze.Rd
View file

@ -24,6 +24,10 @@ a 3-by-1-by-1-by-2 array, then squeeze(A) returns a 3-by-2 matrix. If A is a
row vector, column vector, scalar, or an array with no dimensions of length
1, then squeeze returns the input A.
}
\note{
This is basically a wrapper of drop() with a minor adjustment to adapt
the output to what happens on Matlab
}
\author{
Waldir Leoncio
}