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

Improved handling of extra dimensions

Browse source
This commit is contained in:
Waldir Leoncio 2020-03-18 11:42:41 +01:00
parent 8a73fe4bc1
commit 23e0d4a1a3
3 changed files with 20 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

16
R/zeros_ones.R
View file

@ -24,8 +24,14 @@ zeros_or_ones <- function(n, x) {
#' the `zeros()` function on Matlab
#' @param n1 number of rows
#' @param n2 number of columns
#' @param ... extra dimensions
zeros <- function(n1, n2 = n1, ...) {
return(zeros_or_ones(c(n1, n2, ...), 0))
if (length(n1) == 1) {
n <- c(n1, n2, ...)
} else {
n <- n1
}
return(zeros_or_ones(n, 0))
}
#' @title Matrix of ones
@ -33,6 +39,12 @@ zeros <- function(n1, n2 = n1, ...) {
#' the `ones()` function on Matlab
#' @param n1 number of rows
#' @param n2 number of columns
#' @param ... extra dimensions
ones <- function(n1, n2 = n1, ...) {
return(zeros_or_ones(c(n1, n2, ...), 1))
if (length(n1) == 1) {
n <- c(n1, n2, ...)
} else {
n <- n1
}
return(zeros_or_ones(n, 1))
}