神经网络学习--PyTorch学习01

PyTorch中的数据类型

分为浮点型和整数型两种：

import torch
# 浮点型
float_a = torch.FloatTensor(4, 5)  # size为2*3的矩阵
float_b = torch.FloatTensor([1, 2, 3, 4])  # [1,2,3,4]矩阵

# 整形
int_a = torch.IntTensor(4, 5)
int_b = torch.IntTensor([1, 2, 3, 4])

初始化数组

import torch

# 随机生成浮点型Tensor
rand_a = torch.rand(2, 3)

# 随机生成E=0，D=1 正态分布的Tensor
randn_a = torch.randn(2, 3)

# 设定起始值、结束值、步长 的Tensor 此方法在0.5版本将被更改
range_a = torch.range(1, 20, 1)

# 设定数据为0的Tensor
zeros_a=torch.zeros(2, 3)

　　

 

来源：https://www.cnblogs.com/zuhaoran/p/11452082.html