torch.squeeze
function squeeze<S extends Shape, D extends DType = DType, Dev extends DeviceType = DeviceType>(input: Tensor<S, D, Dev>, options?: SqueezeOptions): Tensor<SqueezeShape<S, undefined>, D, Dev>function squeeze<S extends Shape, Dim extends number, D extends DType = DType, Dev extends DeviceType = DeviceType>(input: Tensor<S, D, Dev>, dim: Dim, options?: SqueezeOptions): Tensor<SqueezeShape<S, Dim>, D, Dev>Removes dimensions of size 1 from the tensor.
Parameters
inputTensor<S, D, Dev>- The input tensor
optionsSqueezeOptionsoptional- Optional settings for squeeze
Returns
Tensor<SqueezeShape<S, undefined>, D, Dev>– Tensor with size-1 dimensions removedExamples
const x = torch.zeros(2, 1, 3, 1);
torch.squeeze(x).shape; // [2, 3]
torch.squeeze(x, 1).shape; // [2, 3, 1]
torch.squeeze(x, { dim: 3 }).shape; // [2, 1, 3]