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.


mlr3torch documentation built on April 4, 2025, 3:03 a.m.