Squeezes a tensor by calling torch::torch_squeeze() with the given dimension dim.
torch::torch_squeeze()
dim
nn_squeeze(dim)
(integer()) The dimension to squeeze.
integer()