tf.raw_ops.CacheDataset | TensorFlow v2.16.1 (original) (raw)

tf.raw_ops.CacheDataset

Stay organized with collections Save and categorize content based on your preferences.

Creates a dataset that caches elements from input_dataset.

View aliases

Compat aliases for migration

tf.compat.v1.raw_ops.CacheDataset

tf.raw_ops.CacheDataset(
    input_dataset,
    filename,
    output_types,
    output_shapes,
    metadata='',
    name=None
)

A CacheDataset will iterate over the input_dataset, and store tensors. If the cache already exists, the cache will be used. If the cache is inappropriate (e.g. cannot be opened, contains tensors of the wrong shape / size), an error will the returned when used.

Args
input_dataset	A Tensor of type variant.
filename	A Tensor of type string. A path on the filesystem where we should cache the dataset. Note: this will be a directory.
output_types	A list of tf.DTypes that has length >= 1.
output_shapes	A list of shapes (each a tf.TensorShape or list of ints) that has length >= 1.
metadata	An optional string. Defaults to "".
name	A name for the operation (optional).

Returns
A Tensor of type variant.