weixin_43698795的博客

Tensors 张量 一个基本概念一、Tensors常用package1.torch.numel>>>In [10]: a = torch.randn(1,2)Out[10]:tensor([[0.2167, 1.3228]])>>> torch.numel(a)22.torch.eyetorch.eye(n, m=None...

一、工具类 utils.py#!/usr/bin/env python# -*- coding:utf-8 -*-'''utils:画loss曲线，随机显示前六张图片，转换为one_hot编码'''import torchfrom matplotlib import pyplot as plt# 画出loss曲线变化图def plot_curve(data): ...

import numpy as np# y = wx + b ===》 模型E(w,b) = Σ(yi-wxi-b)²# 计算总的平均errordef compute_error_for_line_given_points(b, w, points): totalError = 0 # 循环100次 for i in range(0, len(points)): ...