paddlespeech.t2s.modules.transformer.embedding module
Positional Encoding Module.
- class paddlespeech.t2s.modules.transformer.embedding.LegacyRelPositionalEncoding(d_model: int, dropout_rate: float, max_len: int = 5000)[source]
Bases:
PositionalEncodingRelative positional encoding module (old version).
Details can be found in https://github.com/espnet/espnet/pull/2816.
See : Appendix B in https://arxiv.org/abs/1901.02860
- Args:
- d_model (int):
Embedding dimension.
- dropout_rate (float):
Dropout rate.
- max_len (int):
Maximum input length.
Methods
__call__(*inputs, **kwargs)Call self as a function.
add_parameter(name, parameter)Adds a Parameter instance.
add_sublayer(name, sublayer)Adds a sub Layer instance.
apply(fn)Applies
fnrecursively to every sublayer (as returned by.sublayers()) as well as self.buffers([include_sublayers])Returns a list of all buffers from current layer and its sub-layers.
children()Returns an iterator over immediate children layers.
clear_gradients()Clear the gradients of all parameters for this layer.
create_parameter(shape[, attr, dtype, ...])Create parameters for this layer.
create_tensor([name, persistable, dtype])Create Tensor for this layer.
create_variable([name, persistable, dtype])Create Tensor for this layer.
eval()Sets this Layer and all its sublayers to evaluation mode.
extend_pe(x)Reset the positional encodings.
extra_repr()Extra representation of this layer, you can have custom implementation of your own layer.
forward(x)Compute positional encoding. Args: x (Tensor): Input tensor (batch, time, *). Returns: Tensor: Encoded tensor (batch, time, *). Tensor: Positional embedding tensor (1, time, *).
full_name()Full name for this layer, composed by name_scope + "/" + MyLayer.__class__.__name__
load_dict(state_dict[, use_structured_name])Set parameters and persistable buffers from state_dict.
named_buffers([prefix, include_sublayers])Returns an iterator over all buffers in the Layer, yielding tuple of name and Tensor.
named_children()Returns an iterator over immediate children layers, yielding both the name of the layer as well as the layer itself.
named_parameters([prefix, include_sublayers])Returns an iterator over all parameters in the Layer, yielding tuple of name and parameter.
named_sublayers([prefix, include_self, ...])Returns an iterator over all sublayers in the Layer, yielding tuple of name and sublayer.
parameters([include_sublayers])Returns a list of all Parameters from current layer and its sub-layers.
register_buffer(name, tensor[, persistable])Registers a tensor as buffer into the layer.
register_forward_post_hook(hook)Register a forward post-hook for Layer.
register_forward_pre_hook(hook)Register a forward pre-hook for Layer.
set_dict(state_dict[, use_structured_name])Set parameters and persistable buffers from state_dict.
set_state_dict(state_dict[, use_structured_name])Set parameters and persistable buffers from state_dict.
state_dict([destination, include_sublayers, ...])Get all parameters and persistable buffers of current layer and its sub-layers.
sublayers([include_self])Returns a list of sub layers.
to([device, dtype, blocking])Cast the parameters and buffers of Layer by the give device, dtype and blocking.
to_static_state_dict([destination, ...])Get all parameters and buffers of current layer and its sub-layers.
train()Sets this Layer and all its sublayers to training mode.
backward
register_state_dict_hook
- class paddlespeech.t2s.modules.transformer.embedding.PositionalEncoding(d_model, dropout_rate, max_len=5000, dtype='float32', reverse=False)[source]
Bases:
LayerPositional encoding.
- Args:
- d_model (int):
Embedding dimension.
- dropout_rate (float):
Dropout rate.
- max_len (int):
Maximum input length.
- reverse (bool):
Whether to reverse the input position.
- type (str):
dtype of param
Methods
__call__(*inputs, **kwargs)Call self as a function.
add_parameter(name, parameter)Adds a Parameter instance.
add_sublayer(name, sublayer)Adds a sub Layer instance.
apply(fn)Applies
fnrecursively to every sublayer (as returned by.sublayers()) as well as self.buffers([include_sublayers])Returns a list of all buffers from current layer and its sub-layers.
children()Returns an iterator over immediate children layers.
clear_gradients()Clear the gradients of all parameters for this layer.
create_parameter(shape[, attr, dtype, ...])Create parameters for this layer.
create_tensor([name, persistable, dtype])Create Tensor for this layer.
create_variable([name, persistable, dtype])Create Tensor for this layer.
eval()Sets this Layer and all its sublayers to evaluation mode.
extend_pe(x)Reset the positional encodings.
extra_repr()Extra representation of this layer, you can have custom implementation of your own layer.
forward(x)Add positional encoding.
full_name()Full name for this layer, composed by name_scope + "/" + MyLayer.__class__.__name__
load_dict(state_dict[, use_structured_name])Set parameters and persistable buffers from state_dict.
named_buffers([prefix, include_sublayers])Returns an iterator over all buffers in the Layer, yielding tuple of name and Tensor.
named_children()Returns an iterator over immediate children layers, yielding both the name of the layer as well as the layer itself.
named_parameters([prefix, include_sublayers])Returns an iterator over all parameters in the Layer, yielding tuple of name and parameter.
named_sublayers([prefix, include_self, ...])Returns an iterator over all sublayers in the Layer, yielding tuple of name and sublayer.
parameters([include_sublayers])Returns a list of all Parameters from current layer and its sub-layers.
register_buffer(name, tensor[, persistable])Registers a tensor as buffer into the layer.
register_forward_post_hook(hook)Register a forward post-hook for Layer.
register_forward_pre_hook(hook)Register a forward pre-hook for Layer.
set_dict(state_dict[, use_structured_name])Set parameters and persistable buffers from state_dict.
set_state_dict(state_dict[, use_structured_name])Set parameters and persistable buffers from state_dict.
state_dict([destination, include_sublayers, ...])Get all parameters and persistable buffers of current layer and its sub-layers.
sublayers([include_self])Returns a list of sub layers.
to([device, dtype, blocking])Cast the parameters and buffers of Layer by the give device, dtype and blocking.
to_static_state_dict([destination, ...])Get all parameters and buffers of current layer and its sub-layers.
train()Sets this Layer and all its sublayers to training mode.
backward
register_state_dict_hook
- class paddlespeech.t2s.modules.transformer.embedding.RelPositionalEncoding(d_model, dropout_rate, max_len=5000, dtype='float32')[source]
Bases:
LayerRelative positional encoding module (new implementation). Details can be found in https://github.com/espnet/espnet/pull/2816. See : Appendix B in https://arxiv.org/abs/1901.02860
- Args:
- d_model (int):
Embedding dimension.
- dropout_rate (float):
Dropout rate.
- max_len (int):
Maximum input length.
Methods
__call__(*inputs, **kwargs)Call self as a function.
add_parameter(name, parameter)Adds a Parameter instance.
add_sublayer(name, sublayer)Adds a sub Layer instance.
apply(fn)Applies
fnrecursively to every sublayer (as returned by.sublayers()) as well as self.buffers([include_sublayers])Returns a list of all buffers from current layer and its sub-layers.
children()Returns an iterator over immediate children layers.
clear_gradients()Clear the gradients of all parameters for this layer.
create_parameter(shape[, attr, dtype, ...])Create parameters for this layer.
create_tensor([name, persistable, dtype])Create Tensor for this layer.
create_variable([name, persistable, dtype])Create Tensor for this layer.
eval()Sets this Layer and all its sublayers to evaluation mode.
extend_pe(x)Reset the positional encodings.
extra_repr()Extra representation of this layer, you can have custom implementation of your own layer.
forward(x)Add positional encoding. Args: x (Tensor): Input tensor (batch, time, *). Returns: Tensor: Encoded tensor (batch, time, *).
full_name()Full name for this layer, composed by name_scope + "/" + MyLayer.__class__.__name__
load_dict(state_dict[, use_structured_name])Set parameters and persistable buffers from state_dict.
named_buffers([prefix, include_sublayers])Returns an iterator over all buffers in the Layer, yielding tuple of name and Tensor.
named_children()Returns an iterator over immediate children layers, yielding both the name of the layer as well as the layer itself.
named_parameters([prefix, include_sublayers])Returns an iterator over all parameters in the Layer, yielding tuple of name and parameter.
named_sublayers([prefix, include_self, ...])Returns an iterator over all sublayers in the Layer, yielding tuple of name and sublayer.
parameters([include_sublayers])Returns a list of all Parameters from current layer and its sub-layers.
register_buffer(name, tensor[, persistable])Registers a tensor as buffer into the layer.
register_forward_post_hook(hook)Register a forward post-hook for Layer.
register_forward_pre_hook(hook)Register a forward pre-hook for Layer.
set_dict(state_dict[, use_structured_name])Set parameters and persistable buffers from state_dict.
set_state_dict(state_dict[, use_structured_name])Set parameters and persistable buffers from state_dict.
state_dict([destination, include_sublayers, ...])Get all parameters and persistable buffers of current layer and its sub-layers.
sublayers([include_self])Returns a list of sub layers.
to([device, dtype, blocking])Cast the parameters and buffers of Layer by the give device, dtype and blocking.
to_static_state_dict([destination, ...])Get all parameters and buffers of current layer and its sub-layers.
train()Sets this Layer and all its sublayers to training mode.
backward
register_state_dict_hook
- class paddlespeech.t2s.modules.transformer.embedding.ScaledPositionalEncoding(d_model, dropout_rate, max_len=5000, dtype='float32')[source]
Bases:
PositionalEncodingScaled positional encoding module. See Sec. 3.2 https://arxiv.org/abs/1809.08895
- Args:
- d_model (int):
Embedding dimension.
- dropout_rate (float):
Dropout rate.
- max_len (int):
Maximum input length.
- dtype (str):
dtype of param
Methods
__call__(*inputs, **kwargs)Call self as a function.
add_parameter(name, parameter)Adds a Parameter instance.
add_sublayer(name, sublayer)Adds a sub Layer instance.
apply(fn)Applies
fnrecursively to every sublayer (as returned by.sublayers()) as well as self.buffers([include_sublayers])Returns a list of all buffers from current layer and its sub-layers.
children()Returns an iterator over immediate children layers.
clear_gradients()Clear the gradients of all parameters for this layer.
create_parameter(shape[, attr, dtype, ...])Create parameters for this layer.
create_tensor([name, persistable, dtype])Create Tensor for this layer.
create_variable([name, persistable, dtype])Create Tensor for this layer.
eval()Sets this Layer and all its sublayers to evaluation mode.
extend_pe(x)Reset the positional encodings.
extra_repr()Extra representation of this layer, you can have custom implementation of your own layer.
forward(x)Add positional encoding.
full_name()Full name for this layer, composed by name_scope + "/" + MyLayer.__class__.__name__
load_dict(state_dict[, use_structured_name])Set parameters and persistable buffers from state_dict.
named_buffers([prefix, include_sublayers])Returns an iterator over all buffers in the Layer, yielding tuple of name and Tensor.
named_children()Returns an iterator over immediate children layers, yielding both the name of the layer as well as the layer itself.
named_parameters([prefix, include_sublayers])Returns an iterator over all parameters in the Layer, yielding tuple of name and parameter.
named_sublayers([prefix, include_self, ...])Returns an iterator over all sublayers in the Layer, yielding tuple of name and sublayer.
parameters([include_sublayers])Returns a list of all Parameters from current layer and its sub-layers.
register_buffer(name, tensor[, persistable])Registers a tensor as buffer into the layer.
register_forward_post_hook(hook)Register a forward post-hook for Layer.
register_forward_pre_hook(hook)Register a forward pre-hook for Layer.
Reset parameters.
set_dict(state_dict[, use_structured_name])Set parameters and persistable buffers from state_dict.
set_state_dict(state_dict[, use_structured_name])Set parameters and persistable buffers from state_dict.
state_dict([destination, include_sublayers, ...])Get all parameters and persistable buffers of current layer and its sub-layers.
sublayers([include_self])Returns a list of sub layers.
to([device, dtype, blocking])Cast the parameters and buffers of Layer by the give device, dtype and blocking.
to_static_state_dict([destination, ...])Get all parameters and buffers of current layer and its sub-layers.
train()Sets this Layer and all its sublayers to training mode.
backward
register_state_dict_hook