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
PyTorch--Tensor的常用属性
于 2024-09-06 19:41:02 首次发布