Pytorch 基础 -tensor 数据结构

嵌入式视觉

01-14

torch.Tensor 是一种包含单一数据类型元素的多维矩阵，类似于 numpy 的 array。可以使用使用 torch.tensor() 方法将 python 的 list 或序列数据转换成 Tensor 数据，生成的是dtype 默认是 torch.FloatTensor。