開発版ドキュメント · 0.2.3 30c2f8ba · 入門例の検証対象 0.2.3 · 版情報 · 既知の制限

gwexpy.interop.torch_#

Interoperability with PyTorch tensors.

Functions

from_torch(cls, tensor, t0, dt[, unit])

Create a TimeSeries from a PyTorch tensor.

to_torch(series[, device, dtype, ...])

Convert a series to a PyTorch tensor.
gwexpy.interop.torch_.to_torch(series: TimeSeries | Any, device: str | torch.device | None = None, dtype: torch.dtype | None = None, requires_grad: bool = False, copy: bool = False) → torch.Tensor#

Convert a series to a PyTorch tensor.

パラメータ:

  • series (TimeSeries or array-like) -- Input data.

  • device (str or torch.device, optional) -- Target device.

  • dtype (torch.dtype, optional) -- Target dtype.

  • requires_grad (bool, optional) -- Whether to track gradients.

  • copy (bool, optional) -- If True, always copy data; otherwise share memory if possible.

戻り値:

The converted tensor.

戻り値の型:

torch.Tensor

gwexpy.interop.torch_.from_torch(cls: type[T], tensor: torch.Tensor, t0: Any, dt: Any, unit: str | None = None) → T#

Create a TimeSeries from a PyTorch tensor.

パラメータ:

  • cls (type) -- TimeSeries class to instantiate.

  • tensor (torch.Tensor) -- Input tensor.

  • t0 (Quantity or float) -- Start time.

  • dt (Quantity or float) -- Sample interval.

  • unit (str or Unit, optional) -- Data unit.

戻り値:

The created time series.

戻り値の型:

TimeSeries