Pytorch实现transformer并对时间序列预测

利用Pytorch框架实现用transformer通过输入sine预测cos,代码如下:

1. 网络结构

import math
import torch
from torch import nn


class PositionalEncoding(nn.Module):
    def __init__(self, d_model, dropout=0.1, max_len=

你可能感兴趣的:(pytorch,深度学习,transformer)