nn_unsqueeze function

Unsqueeze

Unsqueezes a tensor by calling torch::torch_unsqueeze() with the given dimension dim.

nn_unsqueeze(dim)

Arguments

• dim: (integer(1))

  The dimension to unsqueeze.