polars.Series.to_jax

Series.to_jax(
    device: jax.Device | str_ | None = None,
) → jax.Array

Преобразовать этот Series в массив JAX.

Добавлено в версии 0.20.27.

Предупреждение

В настоящее время эта функциональность считается нестабильной. Она может быть изменена в любой момент без того, чтобы это считалось нарушающим обратную совместимость изменением.

Параметры:

device

Укажите устройство JAX Device, на котором будет создан массив; можно указать строку (например, «cpu», «gpu» или «tpu»), в этом случае устройство будет получено как jax.devices(string)[0]. Для более точного управления можно напрямую передать экземпляр Device. Если указано None, массивы создаются на устройстве по умолчанию.

Примеры

>>> s = pl.Series("x", [10.5, 0.0, -10.0, 5.5])
>>> s.to_jax()
Array([ 10.5,   0. , -10. ,   5.5], dtype=float32)