tabensemb.data.datamodule.DataModule.get_first_tensor_slice#

method

DataModule.get_first_tensor_slice(indices) → Tensor[source]#

Get a slice of the tensor of continuous features.

Parameters:

indices: The indices to make the slice.

Returns:

torch.Tensor: The sliced tensor of continuous feature.