pytorch
猿文档
这个作者很懒,什么都没留下…
展开
-
pytorch(三) 常用类型
Tensors 张量 一个基本概念 一、Tensors常用package 1.torch.numel >>> In [10]: a = torch.randn(1,2) Out[10]: tensor([[0.2167, 1.3228]]) >>> torch.numel(a) 2 2.torch.eye torch.eye(n, m=None...原创 2020-02-12 16:29:56 · 154 阅读 · 0 评论 -
pytorch(二)手写数字笔记
一、工具类 utils.py #!/usr/bin/env python # -*- coding:utf-8 -*- ''' utils:画loss曲线,随机显示前六张图片,转换为one_hot编码 ''' import torch from matplotlib import pyplot as plt # 画出loss曲线变化图 def plot_curve(data): ...原创 2020-02-12 15:15:30 · 201 阅读 · 0 评论 -
pytorch(一)简单回归笔记
import numpy as np # y = wx + b ===》 模型E(w,b) = Σ(yi-wxi-b)² # 计算总的平均error def compute_error_for_line_given_points(b, w, points): totalError = 0 # 循环100次 for i in range(0, len(points)): ...原创 2020-02-12 15:08:03 · 112 阅读 · 0 评论