weixin_44152895的博客

from __future__ import division import torch import torch.nn as nn import torch.nn.functional as F from torch.autograd import Variable import numpy as np from util import * def get_test_input(): img = cv2.imread("dog-cycle-car.png") img =.

import torch #导入包 #下面定义四个整形变量 batch_n=100 #一个批次中输入数据的数量 hidden_layer=100 #定义经过隐藏层后保留的数据特征的个数 input_data=1000 #每个数据包含的数据特征个数 output_data=10 #输出数据，分类结果值 #通过randn（）随机生成的浮点数，取值满足均值为0，方差为1的正太分布。 x = torch.randn(batch_n,input_data) .