torch.Linear(input_size,output_size, bia = True/Flase)

1. 引入

import torch

import torch.nn as nn

2. 一个线性层

mod= nn.Linear(5,3,bias = True) # input 是5 output 是3

3. 大小为5的随机tensor

x = torch.rand(5)

print (x.size())

4. feed x into module

y = mod(x)

5. 查看weight 矩阵和 bias

print ( mod.weight) print(mod.bias)

print (mod.weight.size()) mod.bias.size()

6. 修改weight矩阵

mod.weight [0,0] = 1

mod.weight[1,1] = 2

mod.bias[1] = 2

7. mod2 = nn.Linear(3,4,bias = Fasle)

print ( mod2.weight)

print (mod2.bias)

##########################################

import torch
import torch.nn as nn

mod1 = nn.Linear(3,4,bias = True)
mod2 = nn.Linear(3,5,bias = False)

x2 = torch.Tensor([1,1,1])

mod1.weight[0,1] = 1

x = torch.rand（3）

y = mod1(x)

print(mod1.weight, mod1.bias)

转载于:https://www.cnblogs.com/shaxiaoshaxiao/p/9867844.html

torch.Linear(input_size,output_size, bia = True/Flase)相关推荐

intval0.57100 php_php中0,'',null,false,true,FLASE,TREU,array()的相等恒等学习
//比较值 '' NULL 0 false true FALSE TRUE //相等判断 //'' == NULL == 0 == false (相等) //array() = 0 == NULL = ...
pytorch系列 ---5以 linear_regression为例讲解神经网络实现基本步骤以及解读nn.Linear函数
本文主要讲述最简单的线性回归函数: y = w x + b y=wx+by=wx+b在pytorch的实现,主要包括神经网络实现的基本步骤和nn.Linear的源码解读. 1. nn.Linear 源 ...
pytorch使用torch.nn.Sequential构建网络
以一个线性回归的例子为例: 全部代码 import torch import numpy as npdef get_x_y():x = np.random.randint(0, 50, 300)y_v ...
pytorch利用多个GPU并行计算多gpu
版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明. 本文链接:https://blog.csdn.net/Answer3664/article/d ...
PyTorch Data Parrallel数据并行
PyTorch Data Parrallel数据并行 • 可选择:数据并行处理 • 本文将学习如何用 DataParallel 来使用多 GPU. 通过 PyTorch 使用多个 GPU 非常简单.可 ...
PyTorch 数据并行处理
PyTorch 数据并行处理可选择:数据并行处理(文末有完整代码下载) 本文将学习如何用 DataParallel 来使用多 GPU. 通过 PyTorch 使用多个 GPU 非常简单.可以将模型放 ...
60分钟入门深度学习工具PyTorch
「免费学习 60+ 节公开课:投票页面,点击讲师头像」作者 | Soumith Chintala 中文翻译.注释制作 | 黄海广配置环境:PyTorch 1.3,python 3.7, 主机:显卡 ...
PyTorch 源码解读之分布式训练了解一下？
来源丨商汤学术编辑丨极市平台本文由浅入深讲解 torch.distributed 这一并行计算包的概念,实现细节和应用方式,并带大家快速入门 PyTorch 分布式训练. 0 前言由于大规模 ...
PyTorch：采用sklearn 工具生成这样的合成数据集+利用PyTorch实现简单合成数据集上的线性回归进行数据分析
PyTorch:采用sklearn 工具生成这样的合成数据集+利用PyTorch实现简单合成数据集上的线性回归进行数据分析目录输出结果核心代码输出结果核心代码 #PyTorch:采用skle ...

torch.Linear(input_size,output_size, bia = True/Flase)

torch.Linear(input_size,output_size, bia = True/Flase)相关推荐

最新文章

热门文章