torch_narrow {torch} | R Documentation |
Narrow
Description
Narrow
Usage
torch_narrow(self, dim, start, length)
Arguments
self |
(Tensor) the tensor to narrow |
dim |
(int) the dimension along which to narrow |
start |
(int) the starting dimension |
length |
(int) the distance to the ending dimension |
narrow(input, dim, start, length) -> Tensor
Returns a new tensor that is a narrowed version of input
tensor. The
dimension dim
is input from start
to start + length
. The
returned tensor and input
tensor share the same underlying storage.
Examples
if (torch_is_installed()) {
x = torch_tensor(matrix(c(1:9), ncol = 3, byrow= TRUE))
torch_narrow(x, 1, 1, 2)
torch_narrow(x, 2, 2, 2)
}
[Package torch version 0.13.0 Index]