torch_scalar_tensor {torch} | R Documentation |
Scalar tensor
Description
Creates a singleton dimension tensor.
Usage
torch_scalar_tensor(value, dtype = NULL, device = NULL, requires_grad = FALSE)
Arguments
value |
the value you want to use
|
dtype |
(torch.dtype , optional) the desired data type of returned tensor. Default: if NULL , uses a global default (see torch_set_default_tensor_type ).
|
device |
(torch.device , optional) the desired device of returned tensor. Default: if NULL , uses the current device for the default tensor type (see torch_set_default_tensor_type ). device will be the CPU for CPU tensor types and the current CUDA device for CUDA tensor types.
|
requires_grad |
(bool, optional) If autograd should record operations on the returned tensor. Default: FALSE .
|
[Package
torch version 0.13.0
Index]