内容速览
张量(Tensor)的基本含义
用到的库和框架
张量的创建
通过列表创建张量
通过元组创建张量
将numpy创建的数组转换成张量
二维数组的创建
张量的类型
查看变量的类型
创建固定类型的张量...用到的库和框架
import torch
import numpy as np
张量的创建
通过列表创建张量
# 通过列表创建张量
t = torch.tensor([1, 2])
t
tensor(...[1, 2])
通过元组创建张量
# 通过元组创建张量
torch.tensor((1, 2))
tensor([1, 2])
将numpy创建的数组转换成张量
a = np.array((1, 2))...二维数组的创建
# 用list的list创建二维数组
t2 = torch.tensor([[1, 2], [3, 4]])
t2
tensor([[1, 2],
[3, 4]])
张量的类型...张量转化为数组numpy
t1.numpy()
array([ 1, 2, 3, 4, 5, 6, 7, 8, 9, 10], dtype=int64)
张量转化为列表tolist
t1