torch.unsqueeze
function unsqueeze<S extends Shape, Dim extends number, D extends DType = DType, Dev extends DeviceType = DeviceType>(input: Tensor<S, D, Dev>, options: UnsqueezeOptions & { dim: Dim }): Tensor<UnsqueezeShape<S, Dim>, D, Dev>function unsqueeze<S extends Shape, Dim extends number, D extends DType = DType, Dev extends DeviceType = DeviceType>(input: Tensor<S, D, Dev>, dim: Dim, options?: UnsqueezeOptions): Tensor<UnsqueezeShape<S, Dim>, D, Dev>Adds a dimension of size 1 at the specified position.
Parameters
inputTensor<S, D, Dev>- The input tensor
optionsUnsqueezeOptions & { dim: Dim }- Optional settings for unsqueeze
Returns
Tensor<UnsqueezeShape<S, Dim>, D, Dev>– Tensor with an additional dimension of size 1Examples
const x = torch.tensor([1, 2, 3]);
torch.unsqueeze(x, 0).shape; // [1, 3]
torch.unsqueeze(x, { dim: 1 }).shape; // [3, 1]