Script.shared_tensor¶
- Script.shared_tensor(*, dtype, shape)¶
Allocate a shared tensor.
This instruction allocates a shared tensor with the specified data type, shape, and (optional) layout.
- Parameters:
dtype (DataType) – The data type of the tensor elements.
shape (Sequence[int]) – The shape of the tensor.
- Returns:
ret – The allocated shared tensor.
- Return type:
Notes
Thread group: Can be executed by any sized thread group.