Skip to main content

Ctrl+K

Quick start
Tutorials
Models
Tricks
API

Ctrl+K

Quick start
Tutorials
Models
Tricks
API

Ctrl+K

Section Navigation

layers
models
- auto_config
  - AutoConfig
- auto_model
- autoformer
- base
- bert
  - Bert
  - BertConfig
- deep_ar
  - DeepAR
  - DeepARConfig
- diffusion
- dlinear
  - DLinear
  - DLinearConfig
- gpt
  - GPT
  - GPTConfig
- informer
- nbeats
  - NBeats
  - NBeatsConfig
- patches_tst
- pfn
- rnn
  - Encoder
  - RNN
  - RNN2
  - RNNConfig
- rwkv
- seq2seq
  - DecoderV1
  - DecoderV2
  - Encoder
  - Seq2seq
  - Seq2seqConfig
- tcn
  - Encoder
  - TCN
  - TCNConfig
- tft
  - TFTransformer
  - TFTransformerConfig
- tide
- timesfm
- transformer
- unet
  - Decoder
  - Encoder
  - Unet
  - UnetConfig
- wavenet
  - DecoderV1
  - DecoderV2
  - Encoder
  - WaveNet
  - WaveNetConfig
trainer

API
layers
rwkv_layer

rwkv_layer#

Classes

`ChannelMixing`(args, *kwargs)	TensorFlow RWKV channel mixing
`TimeMixing`(args, *kwargs)	TensorFlow RWKV time mixing

previous

RotaryPositionEmbedding

next

ChannelMixing

This Page

Show Source

© Copyright 2020, Longxing Tan.

Created using Sphinx 8.1.3.

Built with the PyData Sphinx Theme 0.16.1.