Skip to contents

Unsqueeze

Source: R/PipeOpTorchReshape.R
nn_unsqueeze.Rd

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

Usage

nn_unsqueeze(dim)

Arguments

dim

(integer(1))
The dimension to unsqueeze.