torch_narrow: Narrow

In torch: Tensors and Neural Networks with 'GPU' Acceleration

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)
}

torch documentation built on May 29, 2024, 9:54 a.m.