pytorch基础回顾_(torch的定义、构建、运算、转换等)

最新推荐文章于 2023-06-29 14:25:07 发布

DDxuexi

最新推荐文章于 2023-06-29 14:25:07 发布

阅读量318

点赞数

分类专栏： pytorch学习

本文链接：https://blog.csdn.net/DDxuexi/article/details/118610378

版权

本文介绍了PyTorch中矩阵的构建方法，包括empty(), rand(), zeros()函数，以及如何修改数据类型。此外，还详细讨论了tensor的构建、运算，如加法、切片、view()函数，以及in_place操作。最后，阐述了torch与numpy之间的转换，包括torch.numpy()和torch.from_numpy()，并强调两者共享存储空间的特点。" 109118307,9438201,Linux进程管理：查看、终止与杀死,"['Linux', 'shell', '进程管理']

摘要由CSDN通过智能技术生成

import torch
import numpy as np

矩阵torch的构建

1.empty()初始化一个5*3的矩阵

x = torch.empty(5,3)
x # 和视频中讲的不一样了,这里初始化为了全0的矩阵

tensor([[0., 0., 0.],
        [0., 0., 0.],
        [0., 0., 0.],
        [0., 0., 0.],
        [0., 0., 0.]])

2. rand()随机初始化一个5*3矩阵

x = torch.rand(5,3)
x

tensor([[0.9752, 0.2380, 0.2949],
        [0.4990, 0.8772, 0.1745],
        [0.9362, 0.4198, 0.1290],
        [0.8173, 0.5672, 0.5823],
        [0.7380, 0.2954, 0.5019]])

3. zeros()初始化一个全0的5*3矩阵

x = torch.zeros(5,3)
x

tensor([[0., 0., 0.],
        [0., 0., 0.],
        [0., 0., 0.],
        [0., 0., 0.],
        [0., 0., 0.]])

# 查看x矩阵的数据的类型
x.dtype

torch.float32

4. 修改矩阵的数据类型 : dtype=torch.XXX

x = torch.zeros(5,3,dtype=torch.long)
x.dtype

torch.int64

tensor的构建的其他方式

1.直接将一个数据构建成tensor

x = torch

最低0.47元/天解锁文章

DDxuexi

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
pytorch基础回顾_(torch的定义、构建、运算、转换等)

import torchimport numpy as np矩阵torch的构建1.empty()初始化一个5*3的矩阵x = torch.empty(5,3)x # 和视频中讲的不一样了,这里初始化为了全0的矩阵tensor([[0., 0., 0.], [0., 0., 0.], [0., 0., 0.], [0., 0., 0.], [0., 0., 0.]])2. rand()随机初始化一个5*3矩阵x = tor
复制链接

扫一扫

专栏目录