动手学深度学习（pytorch）学习记录1-数据操作[学习记录]

注：本代码在jupyter notebook上运行

import torch

# 用arange创建行向量
x = torch.arange(12)
print(x)

print(x.shape) # 打印张量的形状

x.numel() # 张量的元素个数

# 用reshape函数改变张量形状(元素值不会改变)
X = x.reshape(3, 4) # 改变成3行4列的形状
print(x)# 小写x
print(X)# 大写X

# 也可以用-1调用自动计算维度的功能
X1 = x.reshape(-1, 4)
X2 = x.reshape(3, -1)
print(X1, '\n' ,X2) # reshape的结果和上面一样

# 创建全0张量 指定形状（2，3，4）
torch.zeros((2, 3, 4))

# 创建全1张量 指定形状（2，3，4）
torch.ones((2, 3,