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

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

一、tensor属性类

1.tensor.dim()

查看tensor的维度。

示例：

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

watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjgzMDY5Nw_size_16_color_FFFFFF_t_70

2.tensor.size()

查看tensor的大小。

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

watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjgzMDY5Nw_size_16_color_FFFFFF_t_70 1

3.tensor.shape

查看tensor的形状。

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

watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjgzMDY5Nw_size_16_color_FFFFFF_t_70 2

发表评论

表情：
评论列表 （有 0 条评论，368人围观）

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

相关阅读

    相关 pytorch损失函数

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

    曾经终败给现在曾经终败给现在/ 0 赞/ 494 阅读