# tf.raw_ops.TakeDataset

<br />

Creates a dataset that contains `count` elements from the `input_dataset`.

#### View aliases


**Compat aliases for migration**

See
[Migration guide](https://www.tensorflow.org/guide/migrate) for
more details.

[`tf.compat.v1.raw_ops.TakeDataset`](https://www.tensorflow.org/api_docs/python/tf/raw_ops/TakeDataset)

<br />

    tf.raw_ops.TakeDataset(
        input_dataset,
        count,
        output_types,
        output_shapes,
        metadata='',
        name=None
    )

<br />

<br />

<br />

| ## Args ||
|---|---|
| `input_dataset` | A `Tensor` of type `variant`. |
| `count` | A `Tensor` of type `int64`. A scalar representing the number of elements from the `input_dataset` that should be taken. A value of `-1` indicates that all of `input_dataset` is taken. |
| `output_types` | A list of `tf.DTypes` that has length `>= 1`. |
| `output_shapes` | A list of shapes (each a [`tf.TensorShape`](https://www.tensorflow.org/versions/r2.13/api_docs/python/tf/TensorShape) or list of `ints`) that has length `>= 1`. |
| `metadata` | An optional `string`. Defaults to `""`. |
| `name` | A name for the operation (optional). |

<br />

<br />

<br />

<br />

| ## Returns ||
|---|---|
| A `Tensor` of type `variant`. ||

<br />