目录

魔改测试代码:

Basisformer论文介绍


import torch.nn as nn
import torch.nn.utils.weight_norm as wn
import matplotlib.pyplot as plt
import torch
import torch.nn.functional as F
import time
import math
import numpy as np


class MLP_bottle(nn.Module):
    def __init__(self, input_len, output_len, bottleneck, bias=True):
        super().__init__()
        self.linear1 = nn.Sequential(wn(nn.Linear(input_len, bottleneck, bias=bias)), nn.ReLU(), w

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注