Returns a Bijector variant of tf.nest.pack_sequence_as.
oryx.bijectors.pack_sequence_as(
example, name='restructure'
)
See also the Restructure
bijector for general rearrangements.
Args | |
---|---|
example
|
A Tensor or (potentially nested) collection of Tensors. |
name
|
An optional Python string, inserted into names of TF ops created by this bijector. |
Example
x = tf.constant(1)
example = collections.OrderedDict([
('a', [x, x, x]),
('b', x)])
bij = tfb.pack_sequence_as(example)
bij.forward([1, 2, 3, 4.])
# Returns
# collections.OrderedDict([
# ('a', [1, 2, 3]),
# ('b', 4.)])