PyTorch--Tensor的常用属性

t = torch.tensor([[1, 2, 3], [2, 3, 4]])
print(t)
print("类型：", t.dtype)  # tensor 类型
print("形状：", t.shape)  # tensor 形状
print("形状：", t.size())  # tensor 的形状，相比shape更灵活
print("第1维大小：", t.size(0))  # tensor 第1维大小
print("第2维大小：", t.size(1))  # tensor 第2维大小
print("维度大小：", t.dim())  # tensor 维度大小
print("梯度：", t.grad)  # tensor 梯度
print("所在设备：", t.device)  # tensor 存储的设备
print("否存储在GPU上：", t.is_cuda)  # 张量是否存储在 GPU 上

# 运行结果：
# tensor([[1, 2, 3],
#         [2, 3, 4]])
# 类型： torch.int64
# 形状： torch.Size([2, 3])
# 形状： torch.Size([2, 3])
# 第1维大小： 2
# 第2维大小： 3
# 维度大小： 2
# 梯度： None
# 所在设备： cpu
# 否存储在GPU上： False