2.1 Tensor张量

1、张量包含的属性

dtype：

2. 张量的创建

① 直接进行创建（torch.tensor()）

import numpy as np
import torch

# 创造一个numpy数组
t0 = np.ones((3,3))
# 将数组转化为张量
t1 = torch.tensor(t0)
# 将数组转化为GPU上的张量
t2 = torch.tensor(t0,device = 'cuda')

print('t0=',t0,t0.dtype)
print('t1=',t1)
print('t2=',t2)

结果为：

② 根据ndarray进行创建（torch.from_numpy()）

注意：生成的张量与numpy数组共用内存，改变其中一个另一个也跟着改变

# 转换张量后改变元素
t3 = torch.from_numpy(t0)
t0[0,0] = 3
print('t3=',t3)

结果：

结果表示：t3中的元素随着t0中的元素改变了