pytorch学习笔记1：张量

目录

 

Tensor与numpy相互转换：

torch.ones(),torch.add(),torch.zeros(),torch.squeeze()

---

Tensor与numpy相互转换：

import torch
x=torch.Tensor(2,3)#生成一个4*5的Tensor张量

将Tensor转换为numpy数组
y=x.numpy()

将numpy数组转换为Tensor
import numpy as np
z=torch.from_numpy(x)

torch.ones(),torch.add(),torch.zeros(),torch.squeeze()

torch.ones()

torch.ones(*sizes, out=None) → Tensor

返回一个全为1 的张量，形状由可变参数sizes定义。

参数:

 

• sizes (int...) – 整数序列，定义了输出形状
• out (Tensor, optional) – 结果张量 例子:

>>> torch.ones(2, 3)

 1  1  1
 1  1  1
[torch.FloatTensor of size 2x3]

>>> torch.ones(5)

 1
 1
 1
 1
 1
[torch.FloatTensor of size 5]

 torch.add()<