Pycharm-pytorch基础(1)

import torch
import numpy as np
#1.tensor介绍
#?torch.tensor
a=torch.tensor(1.0,dtype=torch.float)
print(a)

#与numpy中数组与tensor中张量的相互读取
g=np.array([[1,2,3],[4,5,6]])

#数组-》张量
h=torch.tensor(g)
print(h)
i=torch.from_numpy(g)
print(i)
#张量-》数组
j=h.numpy()
print(j)

#构建
k=torch.rand(2,3)
l=torch.ones(2,3)
m=torch.zeros(2,3)
n=torch.arange(0,10,2)
print(k,'\n',l,'\n',m,'\n',n)

#查看tensor维度信息的方式
print(k.shape)
#or
print(k.size())

#tensor可以通过add计算
# alpha * b + a, 维度不够的地方自动扩容
#print(torch.add(a, b, alpha=10))

#tensor索引与numpy类似
print(k[:,1],'\n',k[0,:])

#clone 与copy（感觉没什么区别？）
#https://zhuanlan.zhihu.com/p/393041305

#改变tensor形状view和torch.reshape()
print(k.view(3,2))
print(k.view(-1,2))#二列三行

#torch.view()会改变原张量，如果想原始张量和变换后的张量互相不影响，用torch.reshape()
#或.clone()后用torch.view()

#广播机制，维度不一样会自动配齐
p=torch.arange(1,3).view(1,2)
print(p)
q=t