torch_flatten: Flatten

Description Usage Arguments flatten(input, start_dim=0, end_dim=-1) -> Tensor Examples

View source: R/gen-namespace.R

Description

Flatten

Usage

1
torch_flatten(self, dims, start_dim = 1L, end_dim = -1L, out_dim)

Arguments

self

(Tensor) the input tensor.

dims

if tensor is named you can pass the name of the dimensions to flatten

start_dim

(int) the first dim to flatten

end_dim

(int) the last dim to flatten

out_dim

the name of the resulting dimension if a named tensor.

flatten(input, start_dim=0, end_dim=-1) -> Tensor

Flattens a contiguous range of dims in a tensor.

Examples

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

t = torch_tensor(matrix(c(1, 2), ncol = 2))
torch_flatten(t)
torch_flatten(t, start_dim=2)
}

Example output



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