pytorch入门

使用torch.randn构建正态分布的随机数

import torch 
a=torch.randn(2,3)#使用标准正态分布,随机生成

>>>a
>tensor([1,2,3],
>			[2,3,4])

>>>a.size(0)
>2
>>>a.size(1)
>3
>>>a.shape
>torch.Size([2,3])
>>>a.shape[1]
>3

构建torch.rand()【0，1】均匀分布

a=torch.rand(1,2,3)
>>>a

>>>a[0]

为了和python进行交互，可用list保存

list(a.shape)