torch_tensor | R Documentation |
Converts R objects to a torch tensor
torch_tensor( data, dtype = NULL, device = NULL, requires_grad = FALSE, pin_memory = FALSE )
data |
an R atomic vector, matrix or array |
dtype |
a torch_dtype instance |
device |
a device creted with |
requires_grad |
if autograd should record operations on the returned tensor. |
pin_memory |
If set, returned tensor would be allocated in the pinned memory. |
if (torch_is_installed()) { torch_tensor(c(1, 2, 3, 4)) torch_tensor(c(1, 2, 3, 4), dtype = torch_int()) }
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.