torch can be installed from CRAN with:
install.packages("torch")You can also install the development version with:
remotes::install_github("mlverse/torch")At the first package load additional software will be installed.
If you would like to install with Docker, please read following document.
You can create torch tensors from R objects with the
torch_tensor function and convert them back to R objects
with as_array.
library(torch)
x <- array(runif(8), dim = c(2, 2, 2))
y <- torch_tensor(x, dtype = torch_float64())
y
#> torch_tensor
#> (1,.,.) =
#> 0.7658 0.6123
#> 0.3150 0.4639
#>
#> (2,.,.) =
#> 0.0604 0.0290
#> 0.9553 0.6541
#> [ CPUDoubleType{2,2,2} ]
identical(x, as_array(y))
#> [1] TRUEIn the following snippet we let torch, using the autograd feature, calculate the derivatives:
x <- torch_tensor(1, requires_grad = TRUE)
w <- torch_tensor(2, requires_grad = TRUE)
b <- torch_tensor(3, requires_grad = TRUE)
y <- w * x + b
y$backward()
x$grad
#> torch_tensor
#> 2
#> [ CPUFloatType{1} ]
w$grad
#> torch_tensor
#> 1
#> [ CPUFloatType{1} ]
b$grad
#> torch_tensor
#> 1
#> [ CPUFloatType{1} ]No matter your current skills it’s possible to contribute to
torch development. See the contributing
guide for more information.