import { Tensor } from '@tensorflow/tfjs-core';
import { default as BaseLayer } from './BaseLayer';
import { GPTConfig, ModelForwardAttributes } from '../main';
export default class PositionEmbedding extends BaseLayer {
    private wpe?;
    constructor(config: GPTConfig, name?: string, parent?: BaseLayer);
    forward(attrs: ModelForwardAttributes, x: Tensor): Tensor;
}
