FixedBiasComponent applies a fixed per-element bias; it's similar to the AddShift component in the nnet1 setup (and only needed for nnet1 model conversion. More...
#include <nnet-component.h>
Public Member Functions | |
FixedBiasComponent () | |
virtual std::string | Type () const |
virtual std::string | Info () const |
void | Init (const CuVectorBase< BaseFloat > &scales) |
virtual void | InitFromString (std::string args) |
Initialize, typically from a line of a config file. More... | |
virtual int32 | InputDim () const |
Get size of input vectors. More... | |
virtual int32 | OutputDim () const |
Get size of output vectors. More... | |
virtual void | Propagate (const ChunkInfo &in_info, const ChunkInfo &out_info, const CuMatrixBase< BaseFloat > &in, CuMatrixBase< BaseFloat > *out) const |
Perform forward pass propagation Input->Output. More... | |
virtual void | Backprop (const ChunkInfo &in_info, const ChunkInfo &out_info, const CuMatrixBase< BaseFloat > &in_value, const CuMatrixBase< BaseFloat > &out_value, const CuMatrixBase< BaseFloat > &out_deriv, Component *to_update, CuMatrix< BaseFloat > *in_deriv) const |
Perform backward pass propagation of the derivative, and also either update the model (if to_update == this) or update another model or compute the model derivative (otherwise). More... | |
virtual bool | BackpropNeedsInput () const |
virtual bool | BackpropNeedsOutput () const |
virtual Component * | Copy () const |
Copy component (deep copy). More... | |
virtual void | Read (std::istream &is, bool binary) |
virtual void | Write (std::ostream &os, bool binary) const |
Write component to stream. More... | |
Public Member Functions inherited from Component | |
Component () | |
virtual int32 | Index () const |
Returns the index in the sequence of layers in the neural net; intended only to be used in debugging information. More... | |
virtual void | SetIndex (int32 index) |
virtual std::vector< int32 > | Context () const |
Return a vector describing the temporal context this component requires for each frame of output, as a sorted list. More... | |
void | Propagate (const ChunkInfo &in_info, const ChunkInfo &out_info, const CuMatrixBase< BaseFloat > &in, CuMatrix< BaseFloat > *out) const |
A non-virtual propagate function that first resizes output if necessary. More... | |
virtual | ~Component () |
Protected Member Functions | |
KALDI_DISALLOW_COPY_AND_ASSIGN (FixedBiasComponent) | |
Protected Attributes | |
CuVector< BaseFloat > | bias_ |
Additional Inherited Members | |
Static Public Member Functions inherited from Component | |
static Component * | ReadNew (std::istream &is, bool binary) |
Read component from stream. More... | |
static Component * | NewFromString (const std::string &initializer_line) |
Initialize the Component from one line that will contain first the type, e.g. More... | |
static Component * | NewComponentOfType (const std::string &type) |
Return a new Component of the given type e.g. More... | |
FixedBiasComponent applies a fixed per-element bias; it's similar to the AddShift component in the nnet1 setup (and only needed for nnet1 model conversion.
Definition at line 1542 of file nnet-component.h.
|
inline |
Definition at line 1544 of file nnet-component.h.
|
virtual |
Perform backward pass propagation of the derivative, and also either update the model (if to_update == this) or update another model or compute the model derivative (otherwise).
Note: in_value and out_value are the values of the input and output of the component, and these may be dummy variables if respectively BackpropNeedsInput() or BackpropNeedsOutput() return false for that component (not all components need these).
num_chunks lets us treat the input matrix as contiguous-in-time chunks of equal size; it only matters if splicing is involved.
Implements Component.
Definition at line 3446 of file nnet-component.cc.
|
inlinevirtual |
Reimplemented from Component.
Definition at line 1568 of file nnet-component.h.
|
inlinevirtual |
Reimplemented from Component.
Definition at line 1569 of file nnet-component.h.
References kaldi::cu::Copy().
|
virtual |
Copy component (deep copy).
Implements Component.
Definition at line 3456 of file nnet-component.cc.
References FixedBiasComponent::bias_.
|
virtual |
Reimplemented from Component.
Definition at line 3427 of file nnet-component.cc.
References Component::Info(), and kaldi::VecVec().
void Init | ( | const CuVectorBase< BaseFloat > & | scales | ) |
Definition at line 3407 of file nnet-component.cc.
References CuVectorBase< Real >::Dim(), and KALDI_ASSERT.
Referenced by kaldi::ConvertAddShiftComponent(), and kaldi::nnet2::UnitTestFixedBiasComponent().
|
virtual |
Initialize, typically from a line of a config file.
The "args" will contain any parameters that need to be passed to the Component, e.g. dimensions.
Implements Component.
Definition at line 3412 of file nnet-component.cc.
References AffineComponentPreconditionedOnline::Init(), KALDI_ERR, kaldi::nnet2::ParseFromString(), kaldi::ReadKaldiObject(), and AffineComponentPreconditionedOnline::Type().
|
inlinevirtual |
|
protected |
|
inlinevirtual |
Get size of output vectors.
Implements Component.
Definition at line 1555 of file nnet-component.h.
References Component::Propagate().
|
virtual |
Perform forward pass propagation Input->Output.
Each row is one frame or training example. Interpreted as "num_chunks" equally sized chunks of frames; this only matters for layers that do things like context splicing. Typically this variable will either be 1 (when we're processing a single contiguous chunk of data) or will be the same as in.NumFrames(), but other values are possible if some layers do splicing.
Implements Component.
Definition at line 3438 of file nnet-component.cc.
References CuMatrixBase< Real >::AddVecToRows(), and CuMatrixBase< Real >::CopyFromMat().
|
virtual |
Implements Component.
Definition at line 3470 of file nnet-component.cc.
References kaldi::nnet2::ExpectOneOrTwoTokens(), and kaldi::ExpectToken().
|
inlinevirtual |
Implements Component.
Definition at line 1545 of file nnet-component.h.
|
virtual |
Write component to stream.
Implements Component.
Definition at line 3463 of file nnet-component.cc.
References kaldi::WriteToken().
Definition at line 1575 of file nnet-component.h.
Referenced by FixedBiasComponent::Copy().