ErikTse Runtime

  • 首页 / Home
  • | 算法学习 / Algorithm
    • 所有 / All
    • 简单 / Easy
    • 中等 / Medium
    • 困难 / Hard
  • | 技术分享 / Technology
    • 所有 / All
    • 网络技术 / NetWork
    • 资源共享 / Resource
    • 项目实践 / Event
  • ETOJ在线评测系统
Keep Going.
温故而知新.
  1. 首页
  2. 技术分享
  3. 学科学习
  4. 正文

Python基础语法2(Pytorch)

2022年9月2日 109点热度 0人点赞 0条评论

之前写过《记录PYTHON 3的一些基础语法》,现在来拓展一下知识和强化代码功底。

enumerate()函数

enumerate() 函数用于将一个可遍历的数据对象(如列表、元组或字符串)组合为一个索引序列,同时列出下标和数据,一般用在 for 循环当中。

Python 2.3. 以上版本可用。

lst = ['a', 'b', 'c', 'd', 'e']
for idx, val in enumerate(lst):
    print(idx, val)

'''
output:
0 a
1 b
2 c
3 d
4 e
'''

生成随机tensor矩阵

import torch  # 引入头文件

A = torch.rand([3, 2]) * 5  # 创建出随机的3 x 2的矩阵,大小[0., 5.]
print(A)

随机矩阵通常作为机器学习的初始参数矩阵。

输出tensor矩阵形状

print(A.shape)

输出torch.Size([3, 2])表示这是一个3 x 2的矩阵。

tensor矩阵转置

A = torch.rand([3, 2]) * 5  # 创建出随机的3 x 2的矩阵,大小[0., 5.]
A = A.T
print(A)

或者下面这样写也可以得到转置矩阵

A = torch.t(A)

tensor矩阵乘法

import torch  # 引入头文件

A = torch.rand([3, 2]) * 5  # 创建出随机的3 x 2的矩阵,大小[0., 5.]
B = torch.rand([2, 4]) * 6  # 创建随机的2 x 6的矩阵,大小[0., 6.]
print(torch.matmul(A, B))  # 第一个矩阵的列数等于第二个矩阵的行数,才能相乘

注意第一个矩阵的列数等于第二个矩阵的行数,才能相乘,比如上面的[3, 2]可以和[2, 4]相乘,但是不能和[3, 4]相乘,这是线性代数的知识。

本作品采用 知识共享署名-非商业性使用 4.0 国际许可协议 进行许可
标签: enumerate python tensor torch 基础 语法
最后更新:2022年9月2日

Eriktse

18岁,性别未知,ACM-ICPC现役选手,ICPC亚洲区域赛银牌摆烂人,CCPC某省赛铜牌蒟蒻,武汉某院校计算机科学与技术专业本科在读。

点赞
< 上一篇
下一篇 >

文章评论

取消回复

Eriktse

18岁,性别未知,ACM-ICPC现役选手,ICPC亚洲区域赛银牌摆烂人,CCPC某省赛铜牌蒟蒻,武汉某院校计算机科学与技术专业本科在读。

文章目录
  • enumerate()函数
  • 生成随机tensor矩阵
  • 输出tensor矩阵形状
  • tensor矩阵转置
  • tensor矩阵乘法

友情链接 | 站点地图

COPYRIGHT © 2022 ErikTse Runtime. ALL RIGHTS RESERVED.

Theme Kratos | Hosted In TENCENT CLOUD

赣ICP备2022001555号-1

赣公网安备 36092402000057号