pytorch1.1

1.基本元素操作

import torch

# 创建时可以不指定dtype，默认为long；元素类型可以由精度大的向精度小的转换
# 精度大小：double

2.加法操作

import torch

x=torch.rand(4,3)
y=torch.rand(4,3)
print(x)
print(y)
# 1
print(x+y)
# 2
print(torch.add(x,y))

# 3.设一个空张量（result），把x+y结果输出到result
result=torch.empty(4,3)
torch.add(x,y,out=result)
print(result)

# 4.y.add_(x)，结果存入y中，y的原数据被覆盖
y.add_(x)
print(y)

3.切片操作

import torch

x=torch.rand(6,4)
print(x)

# 提取x前1列
print(x[:,:1])

# 提取x前3列
print(x[:,:3])