jax-js
    Preparing search index...

    Function devicePut

    • Transfer x to device.

      x may be a nested container of arrays or scalars. The resulting structure is committed to the device.

      If device is not specified, this function behaves as identity if the input is already an Array, otherwise it places the scalar uncommitted on the default device.

      Type Parameters

      • T extends unknown

      Parameters

      Returns Promise<MappedJsTree<T, number | boolean, Array>>