Pytorch笔记——2.Pytorch常见函数

红太狼 2022-11-20 05:59 232阅读 0赞

一、tensor属性类

1.tensor.dim()

查看tensor的维度。

示例：

import torch
    import numpy as np
    
    # 定义一个3维向量
    a = torch.tensor(np.random.rand(24).reshape(2,3,4), dtype=torch.float)
    print(a)
    # 打印tensor的维度
    print(a.dim())

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjgzMDY5Nw_size_16_color_FFFFFF_t_70][]

2.tensor.size()

查看tensor的大小。

import torch
    import numpy as np
    
    # 定义一个3维向量
    a = torch.tensor(np.random.rand(24).reshape(2,3,4), dtype=torch.float)
    print(a)
    # 查看tensor的大小
    print(a.size())

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjgzMDY5Nw_size_16_color_FFFFFF_t_70 1][]

3.tensor.shape

查看tensor的形状。

import torch
    import numpy as np
    
    # 定义一个3维向量
    a = torch.tensor(np.random.rand(24).reshape(2,3,4), dtype=torch.float)
    print(a)
    # 查看tensor的形状
    print(a.shape)

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjgzMDY5Nw_size_16_color_FFFFFF_t_70 2][]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjgzMDY5Nw_size_16_color_FFFFFF_t_70]: /images/20221120/8c7dcde3b9fe48dd9dab718066198bce.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjgzMDY5Nw_size_16_color_FFFFFF_t_70 1]: /images/20221120/cfd379ae81134bb08f983644f61594a5.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjgzMDY5Nw_size_16_color_FFFFFF_t_70 2]: /images/20221120/3a4b3545eda14e33b83657ad629c6e16.png

发表评论取消回复

表情：

评论列表（有 0 条评论，232人围观）

还没有评论，来说两句吧...

相关阅读

相关 pytorch神经网络学习笔记(2)

用python搭建了个简单的神经网络关系拟合 import torch from torch.autograd import Variable

妖狐艹你老母/ 2023年02月22日 05:11/ 0 赞/ 115 阅读

相关 Pytorch笔记——常见问题大全

一、torch.tensor和torch.Tensor有什么区别？ 1.torch.tensor是一个函数，返回的是一个tensor，可以通过dtype指定矩阵内数据类型。

落日映苍穹つ/ 2022年11月22日 05:16/ 0 赞/ 152 阅读

相关 Pytorch笔记——1.Pytorch基本数据类型

一、Pytorch基本数据类型在创建tensor时，需要使用torch.Tensor、torch.DoubleTensor等事先指定数据类型。tensor是包含单一数据

蔚落/ 2022年11月22日 05:10/ 0 赞/ 484 阅读

相关 PyTorch：全局函数

torch.randn `torch.``randn`(\size, \, out=None, dtype=None, layout=torch.strided, dev

喜欢ヅ旅行/ 2022年11月20日 13:57/ 0 赞/ 176 阅读

相关 Pytorch笔记——2.Pytorch常见函数

一、tensor属性类 1.tensor.dim() 查看tensor的维度。示例： import torch import numpy as np

红太狼/ 2022年11月20日 05:59/ 0 赞/ 233 阅读

相关【Pytorch学习笔记】2.Pytoch基础

文章目录 4.基本数据类型 4.1 All is about Tensor 4.2 How to denote string

Dear 丶/ 2022年09月04日 05:54/ 0 赞/ 271 阅读

相关 PYTORCH一-what is pytorch 笔记

原文：http://pytorch.org/tutorials/beginner/blitz/tensor\_tutorial.html 声明一个张量　　x =

朴灿烈づ我的快乐病毒、/ 2022年05月28日 12:28/ 0 赞/ 238 阅读

相关 PyTorch，CNN笔记

1. 对于MNIST数据集，如果用全连接层实现的话，大概需要390K参数；用ＣＮＮ实现大概60K参数。基于图片的局部相关性，采用权值共享，大大减少了CNN训练所需的参数！权值

喜欢ヅ旅行/ 2022年03月21日 15:44/ 0 赞/ 254 阅读

相关 pytorch损失函数

> 学习深度学习的过程中一直会遇到损失函数，均方损失函数、交叉熵损失函数、负对数似然损失函数 > 有些时候觉得有点不清晰，所以总结、梳理一下，加深自己的理解 MSELo

曾经终败给现在/ 2021年12月19日 04:49/ 0 赞/ 350 阅读

相关 Pytorch 激活函数

激活函数（activation functions）的目标是，将神经网络非线性化。激活函数是连续的（continuous），且可导的（differential）。连续

一时失言乱红尘/ 2021年12月18日 23:15/ 0 赞/ 323 阅读