View source on GitHub |
Initializer which concats other intializers.
tfp.layers.BlockwiseInitializer( initializers, sizes, validate_args=False ) Args | |
|---|---|
initializers | list of Keras initializers, e.g., "glorot_uniform" or tf.keras.initializers.Constant(0.5413). |
sizes | list of int scalars representing the number of elements associated with each initializer in initializers. |
validate_args | Python bool indicating we should do (possibly expensive) graph-time assertions, if necessary. |
Attributes | |
|---|---|
initializers | |
sizes | |
validate_args | |
Methods
from_config
@classmethodfrom_config( config )
Instantiates an initializer from a configuration dictionary.
get_config
get_config() Returns initializer configuration as a JSON-serializable dict.
__call__
__call__( shape, dtype=None ) Returns a tensor object initialized as specified by the initializer.
| Args | |
|---|---|
shape | Shape of the tensor. |
dtype | Optional dtype of the tensor. If not provided will return tensor of tf.float32. |
View source on GitHub