pytorch创建tensor

目录

  • [1. 从numpy创建](#1. 从numpy创建)
  • [2. 从list创建](#2. 从list创建)
  • [3. 创建未初始化tensor](#3. 创建未初始化tensor)
  • [4. 设置默认tensor创建类型](#4. 设置默认tensor创建类型)
  • [5. rand/rand_like, randint](#5. rand/rand_like, randint)
  • [6. randn生成正态分布随机数](#6. randn生成正态分布随机数)
  • [7. full](#7. full)
  • [8. arange/range](#8. arange/range)
  • [9. linspace/logspace](#9. linspace/logspace)
  • [10. Ones/zeros/eye](#10. Ones/zeros/eye)
  • [11. randperm](#11. randperm)

1. 从numpy创建

2. 从list创建

3. 创建未初始化tensor

python 复制代码
Torch.empty()
Torch.FloatTensor(d1, d2, d3)
Torch.IntTensr(d1, d2, d3)

创建未初始化tensor使用时一定要赋值,否则可能报错。

4. 设置默认tensor创建类型

不设置的话默认FloatTensor

5. rand/rand_like, randint

6. randn生成正态分布随机数

mean是均值,std是方差

7. full

8. arange/range

9. linspace/logspace

函数torch.logspace()返回一阶步张量的一维张量,与之间的底数成对数间隔

输出张量是尺寸步长的一维。

用法:torch.logspace(start, end, steps=100, base=10, out=None)

参数:

start:点集的起始值。

end:点集的最终值

steps:在开始和结束之间要采样的点数。默认值:100

base:对数函数的基数,可以是2,10,e。默认值:10.0

out(Tensor, optional):输出张量

10. Ones/zeros/eye


11. randperm

random.shuffle 随机shuffle

可用于随机生成tensor的index,随机取tensor数据

相关推荐
勾股导航2 分钟前
K-means
人工智能·机器学习·kmeans
liliangcsdn2 分钟前
Diff2Flow中扩散和流匹配的对齐探索
人工智能
HAPPY酷3 分钟前
C++ 和 Python 的“容器”对决:从万金油到核武器
开发语言·c++·python
SmartBrain7 分钟前
战略洞察:以AI为代表的第四次工业革命
人工智能·语言模型·aigc
一个处女座的程序猿17 分钟前
AI之Agent之VibeCoding:《Vibe Coding Kills Open Source》翻译与解读
人工智能·开源·vibecoding·氛围编程
Jay Kay23 分钟前
GVPO:Group Variance Policy Optimization
人工智能·算法·机器学习
风指引着方向34 分钟前
归约操作优化:ops-math 的 Sum/Mean/Max 实现
人工智能·wpf
机器之心34 分钟前
英伟达世界模型再进化,一个模型驱动所有机器人!机器人的GPT时刻真正到来
人工智能·openai
纯爱掌门人40 分钟前
终焉轮回里,藏着 AI 与人类的答案
前端·人工智能·aigc
人工智能AI技术44 分钟前
Transformer:大模型的“万能骨架”
人工智能