tirank.Visualization.create_tensor

tirank.Visualization.create_tensor(data_matrix)[source]

Converts a numpy array or list-like object to a float32 PyTorch tensor.

Parameters:

data_matrix (np.ndarray or list) – The input data matrix.

Returns:

The data converted to a float32 PyTorch tensor.

Return type:

torch.Tensor