首先需要先引入相关库
# 导入相关库 pip install torch torchvision matplotlib -i https://pypi.tuna.tsinghua.edu.cn/simple/
注:torch.n是一个Python模块,它包含一系列用于构建神经网络的函数和类。它可以用来定义层,激活函数,损失函数,优化器,初始化器,以及其他神经网络相关的函数和类。 torch.n模块是在PyTorch 1.0版本中引入的。如果你想使用torch.n模块,你需要升级你的PyTorch版本到1.0或更高版本,PyTorch要求Python版本为3.6或更高版本。
# import torch # import torchvision # print(torch.__version__) # print(torchvision.__version__) """此代码可以查看相关库的版本"""
# 导入相关库
import torch
import torch.n as n
import torch.optim as optim
import torch.n.functional as F
from torch.utils.data import DataLoader
from torchvision import datasets, transforms
import matplotlib.pyplot as plt
import numpy as np
# 定义生成器
class Generator(n.Module):
def __init__(self, latent_dim, img_shape):
super(Generator, self).__init__()