torch_take: Take

Description Usage Arguments take(input, index) -> Tensor Examples

View source: R/gen-namespace.R

Description

Take

Usage

1
torch_take(self, index)

Arguments

self

(Tensor) the input tensor.

index

(LongTensor) the indices into tensor

take(input, index) -> Tensor

Returns a new tensor with the elements of input at the given indices. The input tensor is treated as if it were viewed as a 1-D tensor. The result takes the same shape as the indices.

Examples

1
2
3
4
5
if (torch_is_installed()) {

src = torch_tensor(matrix(c(4,3,5,6,7,8), ncol = 3, byrow = TRUE))
torch_take(src, torch_tensor(c(1, 2, 5), dtype = torch_int64()))
}

torch documentation built on Oct. 7, 2021, 9:22 a.m.