torch_tensor function

Converts R objects to a torch tensor

torch_tensor(
  data,
  dtype = NULL,
  device = NULL,
  requires_grad = FALSE,
  pin_memory = FALSE
)

Arguments

• data: an R atomic vector, matrix or array
• dtype: a torch_dtype instance
• device: a device creted with torch_device()
• requires_grad: if autograd should record operations on the returned tensor.
• pin_memory: If set, returned tensor would be allocated in the pinned memory.

Examples

if (torch_is_installed()) {
torch_tensor(c(1, 2, 3, 4))
torch_tensor(c(1, 2, 3, 4), dtype = torch_int())
}