用 PyTorch 库创建了一个随机张量,并演示了多种张量取整和分解操作

张量创建与初始化

a = torch.rand(2,2)

生成一个 2x2 的随机张量,元素值均匀分布在 [0,1) 区间。例如可能输出:

复制代码
tensor([[0.1234, 0.5678],
        [0.9012, 0.3456]])

a = a * 10

将张量每个元素乘以 10,使值域变为 [0,10)。例如:

复制代码
tensor([[1.234, 5.678],
        [9.012, 3.456]])

取整运算

print(torch.floor(a))

向下取整(地板函数),输出不大于原数的最大整数。示例结果:

复制代码
tensor([[1., 5.],
        [9., 3.]])

print(torch.ceil(a))

向上取整(天花板函数),输出不小于原数的最小整数。示例结果:

复制代码
tensor([[2., 6.],
        [10., 4.]])

print(torch.round(a))

四舍五入到最接近的整数。示例结果:

复制代码
tensor([[1., 6.],
        [9., 3.]])

print(torch.trunc(a))

截断小数部分(向零取整)。示例结果:

复制代码
tensor([[1., 5.],
        [9., 3.]])

分解运算

print(torch.frac(a))

提取小数部分。示例结果:

复制代码
tensor([[0.234, 0.678],
        [0.012, 0.456]])

print(a%2)

计算每个元素除以 2 的余数。示例结果:

复制代码
tensor([[1.234, 1.678],
        [1.012, 1.456]])

关键点说明

  • 所有操作都是逐元素(element-wise)进行的
  • 原始张量 a 不会被修改,每个操作都会返回新张量
  • 取整运算返回浮点型张量,但值为整数形式
  • frac% 运算结果保持原始张量的数值精度
相关推荐
Eloudy1 天前
一键构建 pytorch cpp lib 前端和 wheel
人工智能·pytorch·gpu
FriendshipT1 天前
Ultralytics:解读 YOLO26 知识蒸馏
人工智能·pytorch·python·深度学习·yolo
一碗白开水一2 天前
入门实践工程十:PyTorch 模型量化与 ONNX 导出~附:安装依赖库及工程源码
人工智能·pytorch·python·rnn·深度学习·机器学习
hhzz2 天前
《深度学习框架PyTorch入门与实践》系列:04-nn模块详解:用积木搭建神经网络
pytorch·深度学习·神经网络
阳光开朗男孩2 天前
Pytorch的安装与配置
人工智能·pytorch·python
等一朵映山红2 天前
动态图 vs 静态图:PyTorch 与 TensorFlow 架构底层对比
人工智能·pytorch·python
老马聊技术3 天前
Pytorch深度学习环境配置与测试
人工智能·pytorch·python
hhzz3 天前
《深度学习框架PyTorch入门与实践》系列:03-autograd自动微分:反向传播的引擎
人工智能·pytorch·深度学习
青春不败 177-3266-05203 天前
CNN-Transformer:PyTorch遥感【含无人机】影像的分类、检测、语义分割和点云分类
pytorch·目标检测·cnn·transformer·生态学·遥感·地物分类
Lee_jerome3 天前
python神经网络编程入门(二十七)——RNN IMBD搭建情感分类器与基础训练
pytorch·rnn·深度学习·gru·词嵌入·情感分析·imdb