reshape

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

Change the shape of input.

Parameters:
Returns:

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