nn_squeeze: Squeeze

nn_squeezeR 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.


mlr-org/mlr3torch documentation built on April 17, 2025, 8:22 p.m.