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数据

相关推荐
用户42615577610几秒前
Linux服务器排障实战:从CPU飙高到内存泄漏的排查套路
python
Fairy要carry几秒前
实习01-MindSpore 做了什么
人工智能
向哆哆1 分钟前
智能交通目标检测数据集(完整标注+可直接训练YOLO专用版本)
人工智能·yolo·目标检测
陈广亮1 分钟前
AI Agent 记忆系统实战:OpenClaw Memory 最佳实践
人工智能
小雨中_5 分钟前
4.1 Megatron-LM:千卡级集群预训练的“硬核”框架
人工智能·python·深度学习·机器学习·llama
星空椰5 分钟前
FastAPI 进阶:中间件、依赖注入与 ORM
python·fastapi
王解6 分钟前
MetaGPT深度解析:当AI智能体学会“像人一样协作”
网络·人工智能·ai agent
肾透侧视攻城狮17 分钟前
【效率革命】《TensorFlow分布式训练:攻克内存瓶颈与通信延迟的实战方案》
人工智能·深度学习·tensorflow分布式训练·分布式策略·数据/模型并行·多机配置/自定义训练循环·内存不足/设备间通信瓶颈
高洁0119 分钟前
多模态大模型的统一表征与推理范式
人工智能·python·深度学习·机器学习·transformer
啊阿狸不会拉杆20 分钟前
《计算机视觉:模型、学习和推理》第 8 章-回归模型
人工智能·python·学习·机器学习·计算机视觉·回归·回归模型