nn_unsqueeze {mlr3torch}R Documentation

Unsqueeze

Description

Unsqueezes a tensor by calling torch::torch_unsqueeze() with the given dimension dim.

Usage

nn_unsqueeze(dim)

Arguments

dim

(integer(1))
The dimension to unsqueeze.

[Package mlr3torch version 0.1.0 Index]