Squeeze

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

Usage

nn_squeeze(dim)

Arguments

dim

(integer())
The dimension to squeeze.