pytorch使用dataLoader加载自己的numpy矩阵数据

import numpy as np
from torch.utils.data import Dataset,DataLoader,TensorDataset
import math
import os
import torch
import torch.nn as nn
import torch.nn.functional as F
from torchvision import datasets, transforms
import torch.utils.data
import matplotlib.pyplot as plt
#加载csv文件，直接就是矩阵
with open("./data/be