torch.js has not been released yet.

torch.Tensor.Tensor.unsqueeze_

Tensor.unsqueeze_(dim: number): thisTensor.unsqueeze_(options: UnsqueezeOptions): this

Inserts a dimension of size one at the specified position in-place.

Parameters

dimnumber: Position where the dimension should be inserted

Returns

this– This tensor with the new dimension

Examples

const x = torch.zeros(3, 4);
x.unsqueeze_(0);  // Now shape is [1, 3, 4]

See Also

PyTorch tensor.unsqueeze_()

Tensor.unsqueeze