torch_empty_strided {torch} | R Documentation |
Empty_strided
Description
Empty_strided
Usage
torch_empty_strided(
size,
stride,
dtype = NULL,
layout = NULL,
device = NULL,
requires_grad = FALSE,
pin_memory = FALSE
)
Arguments
size |
(tuple of ints) the shape of the output tensor |
stride |
(tuple of ints) the strides of the output tensor |
dtype |
( |
layout |
( |
device |
( |
requires_grad |
(bool, optional) If autograd should record operations on the returned tensor. Default: |
pin_memory |
(bool, optional) If set, returned tensor would be allocated in the pinned memory. Works only for CPU tensors. Default: |
empty_strided(size, stride, dtype=NULL, layout=NULL, device=NULL, requires_grad=False, pin_memory=False) -> Tensor
Returns a tensor filled with uninitialized data. The shape and strides of the tensor is
defined by the variable argument size
and stride
respectively.
torch_empty_strided(size, stride)
is equivalent to
torch_empty(size).as_strided(size, stride)
.
Warning
More than one element of the created tensor may refer to a single memory location. As a result, in-place operations (especially ones that are vectorized) may result in incorrect behavior. If you need to write to the tensors, please clone them first.
Examples
if (torch_is_installed()) {
a = torch_empty_strided(list(2, 3), list(1, 2))
a
a$stride(1)
a$size(1)
}