Script.reshape_shared

Script.reshape_shared

Script.reshape_shared(tensor, shape)

Reshape a shared tensor.

Parameters:
  • tensor (SharedTensor) – The shared tensor to reshape.

  • shape (Sequence[int]) – The new shape of the shared tensor.

Returns:

ret – The reshaped shared tensor.

Return type:

SharedTensor

Notes

  • Thread group: Can be executed by any sized thread group.