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中的元素改变了