nn_squeeze function

Squeeze

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

nn_squeeze(dim)

Arguments

• dim: (integer())

  The dimension to squeeze.