JaxBackend.from_sharedarr#

JaxBackend.from_sharedarr(arr)[source]#

Returns an array of given shape and dtype from shared memory location.

Parameters:
shapetuple

Tuple of integers specifying the shape of the array.

dtypestr

String specifying the dtype of the array.

shmshared_memory.SharedMemory

Shared memory object where the array is stored.

Returns:
BackendArray

Array of the specified shape and dtype from the shared memory location.