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

红太狼 2022-11-20 05:59 300阅读 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 条评论，300人围观）

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

相关阅读

相关 pytorch常见错误

[https://blog.csdn.net/qq\_38156052/article/details/81293412\\_13][https_blog.csdn.net_q

布满荆棘的人生/ 2023年06月04日 02:54/ 0 赞/ 20 阅读

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

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

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

相关【pytorch笔记】损失函数nll_loss

文章目录使用场景函数理解例子演示总结使用场景在用`pytorch`做训练、测试时经常要用到损失函数计算输出与目标结果的差距，例

爱被打了一巴掌/ 2023年02月16日 01:56/ 0 赞/ 34 阅读

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

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

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

相关 PyTorch：全局函数

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

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

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

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

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

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

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

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

相关 PYTORCH一-what is pytorch 笔记

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

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

相关 pytorch损失函数

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

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

相关 Pytorch 激活函数

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

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