tests/testthat/test_tensor_reshape.R

In rTorch: R Bindings to 'PyTorch'

library(testthat)

source("helper_utils.R")

skip_if_no_torch()

# reshape tensor ---------------------------------------------------------------
context("reshape tensor")

test_that("reshape() 1D to 2D", {
    a <-  torch$arange(4.)
    expect_equal(tensor_ndim(a), 1)   # 1D
    expect_equal(tensor_ndim(torch$reshape(a, c(2L, 2L))), 2)  # 2D
})

test_that("reshape()", {
    b <- torch$tensor(list(list(0, 1), list(2, 3)))
    expect_equal(tensor_ndim(b), 2)   # 2D
    expect_error(torch$reshape(b, list(-1L,0L)))
    expect_error(torch$reshape(b, c(-1L,0L)))
    expect_error(torch$reshape(b, c(-1L,)))
})