用 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 和 % 运算结果保持原始张量的数值精度
相关推荐
田里的水稻3 小时前
FA_CNN卷积神经网络十二_CIFAR10全流程PyTorch
人工智能·pytorch·深度学习·神经网络·cnn
论文复现现场6 小时前
RTX 3090 云GPU租赁价格对比:8家平台配置、计费方式与环境测试
linux·运维·人工智能·pytorch·深度学习·云计算·3090
liwulin05066 小时前
【PYTORCH】迷你DECODER-ONLY模型
pytorch·python·深度学习
田里的水稻1 天前
FA_CNN卷积神经网络四---线性层及其拉平PyTorch
人工智能·pytorch·python·深度学习·神经网络·cnn
GPU实战笔记1 天前
NVIDIA L40S 云端任务验证:训练、推理与生成工作流怎样先试再租
pytorch·ai生成·gpu租用·nvidia l40s
QN1幻化引擎1 天前
信号场注意力(SFA):把 KV Cache 变成 O(1) 的双通道注意力——0.1B/0.3B 受控 A/B 精度反升 15.5% / 24.3%
pytorch·深度学习·大模型·transformer·注意力机制·推理优化
田里的水稻1 天前
FA_CNN卷积神经网络二---PyTorch
人工智能·pytorch·python·深度学习·cnn
lucas_AI1 天前
Google 和 Meta 把 TPU 变成 PyTorch「原生设备」:TorchTPU 开源,device="cuda" 改一个词就能上 TPU
pytorch
Time_Memory_cici1 天前
在Windows系统安装Pycharm以及PyTorch方法
pytorch·windows·pycharm
田里的水稻2 天前
FA_CNN卷积神经网络七---优化器PyTorch
人工智能·pytorch·深度学习·神经网络·cnn