用pytorch写一段代码：以西储大学CWRU轴承故障的振动信号数据库作为模型的训练集，故障类型标签包括：正常、内圈故障、外圈故障和滚动体故障四类。首先将一维故障信号转换为二维灰度图作为CNN模型的输...

import torch import torch.nn as nn from torch.utils.data import DataLoader# 加载CWRU轴承故障训练数据 cwru_dataset = load_cwru_dataset()# 将一维信号转换为二维灰度图 def convert_1d_to_2d(data): # ...# 将数据集中的一维信号转换为二维灰度图 cwru_dataset = cwru_dataset.map(convert_1d_to_2d)# 定义CNN网络 class CNN(nn.Module): def init(se