More specifically, given [F_0, F_1, ... F_n] which are scalar or vector bijectors this bijector creates a transformation which operates on the vector [x_0, ... x_n] with the transformation [F_0(x_0), F_1(x_1) ..., F_n(x_n)] where x_0, ..., x_n are blocks (partitions) of the vector.

tfb_blockwise(
  bijectors,
  block_sizes = NULL,
  validate_args = FALSE,
  name = NULL
)

Arguments

bijectors

A non-empty list of bijectors.

block_sizes

A 1-D integer Tensor with each element signifying the length of the block of the input vector to pass to the corresponding bijector. The length of block_sizes must be be equal to the length of bijectors. If left as NULL, a vector of 1's is used.

validate_args

Logical indicating whether arguments should be checked for correctness.

name

String, name given to ops managed by this object. Default: E.g., tfb_blockwise(list(tfb_exp(), tfb_softplus()))$name == 'blockwise_of_exp_and_softplus'.

Value

a bijector instance.

See also