Pytorch基础代码

1. 创建tensor的方法

import torch

x = torch.empty(5, 3)  # 创建未初始化的矩阵
x = torch.rand(5, 3)  # 创建随机初始化矩阵
x = torch.zeros(5, 3, dtype=torch.long)  # 创建 0 矩阵
x = torch.tensor([5.5, 3])  # 直接从数据构造张量
x = x.new_ones(5, 3, dtype=torch.double)  # he returned Tensor has the same `torch.dtype` `torch.device` as this tensor.
x = torch.randn_like(x, dtype=torch.float)  # 重载 dtype!    
x.size()  # 获取张量形状

2. 运算

y = torch.rand(5, 3)  # 加法形式 1
print(torch.add(x, y))  # 加法形式 2
result = torch.empty(5, 3)  
torch.add(x, y, out=result)  # 加法形式 2+
y.add_(x)  # 加法形式 3
x[:, 1]  # 切片操作
x = torch.randn(4, 4)
y = x.view(16)  # 改变形状
z = x.view(-1, 8)  # -1 自动计算维度大小 
x = torch.randn(1)
x.item()  # item() 方法获取 python 数值

3. 与Numpy转换

貔貅好久沒帶瞭,可以再帶嗎? - 2to在線問答

4年前老公在北京買瞭個玉的貔貅給我''就直接帶脖子上瞭!當時也不知道什麼開光,請啊什麼的就帶瞭大半年!後來懷孕瞭就給老爸保管瞭!現在想在拿出來帶!可以嗎?是不是要開光什麼的!

貔貅吊墜不戴瞭怎麼存放 - 2to在線問答

貔貅手鏈不戴瞭怎麼辦? - 2to在線問答

朋友送瞭我一個他自己戴過的貔貅 我不戴 我該怎麼處理 可不可以直接還給他 還是要怎麼辦? - 2to在線問答

我有一個玉貔貅好長時間不戴瞭!不想要瞭怎麼處理? - 2to在線問答

我有一個玉貔貅好長時間不戴瞭!不想要瞭怎麼處理?

請回來的戴貔貅不適合佩戴要怎麼辦 - 2to在線問答

請回來的戴貔貅不適合佩戴要怎麼辦

貔貅不戴瞭怎麼處理? - 2to在線問答

  1. «
  2. 1 ...
  3. 94
  4. 95
  5. 96
  6. 97
  7. 98
  8. 99
  9. 100

網站簡介

a = torch.ones(5)
b = a.numpy()  # tensor 转换为 numpy
a.add_(1)  # a 改变,b 也会跟着变

import numpy as np

a = np.ones(5)
b = torch.from_numpy(a)  # numpy 转换为 pytorch
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值