polars.Series.to_torch

Series.to_torch() → torch.Tensor

Преобразует этот Series в тензор PyTorch.

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

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

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

Примечания

Тензоры PyTorch не поддерживают UInt16, UInt32 и UInt64; эти типы данных будут автоматически преобразованы соответственно в Int32, Int64 и Int64.

Примеры

>>> s = pl.Series("x", [1, 0, 1, 2, 0], dtype=pl.UInt8)
>>> s.to_torch()
tensor([1, 0, 1, 2, 0], dtype=torch.uint8)
>>> s = pl.Series("x", [5.5, -10.0, 2.5], dtype=pl.Float32)
>>> s.to_torch()
tensor([  5.5000, -10.0000,   2.5000])