baybe.utils.dataframe.to_tensor

baybe.utils.dataframe.to_tensor(*x: ndarray | DataFrame)[source]

Convert numpy arrays and pandas dataframes to tensors.

Parameters:

*x (ndarray | DataFrame) – The array(s)/dataframe(s) to be converted.

Return type:

Tensor | tuple[Tensor, ...]

Returns:

The provided array(s)/dataframe(s) represented as tensor(s).