0
点赞
收藏
分享

微信扫一扫

PyTorch入门基础串讲

南陵王梁枫 2022-04-17 阅读 53
深度学习

PyTorch的基本概念

Tensor

有时遇到的数据是很高维度的,很难使用矩阵或向量来描述,这时就用张量来描述任意维度的物体
Tensor与机器学习的关系

Tensor与机器学习的关系

我们使用tensor来对样本和模型进行描述

Tensor的类型

image-20210318181810970

Tensor的创建

在这里插入图片描述

Tensor创建编程实例

import torch
import numpy as np

"""
几种特殊的tensor
"""
a = torch.Tensor([[1, 2], [3, 4]])
print(a)
print(a.type())
"""
输出:
    tensor([[1., 2.],
            [3., 4.]])
    torch.FloatTensor
"""
#对比
a1 = np.array([[1, 2], [3, 4]])
print(a1)
&
举报

相关推荐

0 条评论