oneflow.unsqueeze¶
-
oneflow.
unsqueeze
(input, dim) → Tensor¶ Returns a new tensor with a dimension of size one inserted at the specified position.
The returned tensor shares the same underlying data with this tensor.
A
dim
value within the range [-input.ndimension() - 1, input.ndimension() + 1) can be used. Negativedim
will correspond tounsqueeze()
applied atdim
=dim + input.ndimension() + 1
.- Parameters
input (Tensor) – the input tensor.
dim (int) – the index at which to insert the singleton dimension
For example:
>>> import numpy as np >>> import oneflow as flow >>> x = flow.randn(2, 3, 4) >>> y = x.unsqueeze(2) >>> y.shape oneflow.Size([2, 3, 1, 4])