nn_squeeze {mlr3torch}R Documentation

Squeeze

Description

Squeezes a tensor by calling torch::torch_squeeze() with the given dimension dim.

Usage

nn_squeeze(dim)

Arguments

dim

(integer())
The dimension to squeeze.

[Package mlr3torch version 0.1.0 Index]