peek

airt.keras.experiments.peek(ds: tf.data.Dataset) -> tf.Tensor

Returns the first element of the dataset

Parameters:

Name	Type	Description	Default
ds	tf.data.Dataset	dataset	required

Returns:

Type	Description
tf.Tensor	the first element of the dataset

Source code in airt/keras/experiments.py

def peek(ds: tf.data.Dataset) -> tf.Tensor:
    """Returns the first element of the dataset

    Args:
        ds: dataset

    Returns:
        the first element of the dataset
    """
    for x in ds:
        return x