Module: tf_agents.utils.composite

Utilities for dealing with CompositeTensors.

Functions

expand_dims(...): Add a new dimension to tensor t along axis.

reshape(...): Reshape composite tensor t to shape.

shape(...)

slice_from(...): Slice a composite tensor along axis from start.

slice_to(...): Slice a composite tensor along axis from 0 to end.

squeeze(...): Squeeze composite tensor along axis axis.