reshape¶

dragon.vm.torch.reshape(
  input,
  shape,
  out=None
)[source]¶

Change the shape of input.

Parameters:

• input (dragon.vm.torch.Tensor) – The input tensor.
• shape (Sequence[int]) – The new shape.
• out (dragon.vm.torch.Tensor, optional) – The optional output tensor.

Returns:

dragon.vm.torch.Tensor – The new tensor.