1、创建张量:
2、张量的方法和属性:
(1) t1.item()只能用于张量中只有一个数据的情况下获取张量内的值
(2)numpy() 将tensor转换为numpy数组
t2 = torch.tensor([[[1, 2]]]) print(t2) print(t2.numpy())
输出结果:
(3)size() 获取该维度里的值
t2 = torch.tensor([[[1, 2]]]) print(t2.size(0))

1、创建张量:
2、张量的方法和属性:
(1) t1.item()只能用于张量中只有一个数据的情况下获取张量内的值
(2)numpy() 将tensor转换为numpy数组
t2 = torch.tensor([[[1, 2]]]) print(t2) print(t2.numpy())
输出结果:
(3)size() 获取该维度里的值
t2 = torch.tensor([[[1, 2]]]) print(t2.size(0))