PyTorch_张量基本运算

基本运算中,包括add, sub, mul, div, neg等函数,以及这些函数的带下划线的版本add_, sub_, mul_, div_, neg_, 其中带下划线的版本为修改原数据。


代码

python 复制代码
import torch 
import numpy as np 

# 不修改原数据的计算
def test01():
    data = torch.randint(0, 10, [2, 3])  # 开始值,结束值,形状
    print(data)

    # 计算完成之后,会返回一个新的张量
    data = data.add(10)
    print(data)

    # data.sub() 
    # data.mul()
    # data.div()
    # data.neg()  取相反数

# 修改原数据的计算 (inplace方式的计算)
def test02():
    data = torch.randint(0, 10, [2, 3])  # 开始值,结束值,形状
    print(data)

    # 带下划线的版本的函数直接修改原数据,不需要用新的变量保存
    data.add_(10)  # inplace=True
    print(data)

    # data.sub_() 
    # data.mul_()
    # data.div_()
    # data.neg_()  取相反数


if __name__ == "__main__":
    test02()
    
相关推荐
纪伊路上盛名在6 分钟前
如何批量获取蛋白质序列的所有结构域(domain)数据-2
数据库·人工智能·机器学习·统计·计算生物学·蛋白质
Actinen6 分钟前
【Python】——注释
python
这张生成的图像能检测吗10 分钟前
(论文速读)InteractVLM: 基于2D基础模型的3D交互推理
人工智能·计算机视觉·交互·生成模型·图像生成·视觉语言模型·3d重建
浣熊-论文指导27 分钟前
人工智能与生物医药融合六大创新思路
论文阅读·人工智能·深度学习·计算机网络·机器学习
自动化小秋葵32 分钟前
Python入门经典题目
开发语言·python
文火冰糖的硅基工坊33 分钟前
[人工智能-大模型-48]:模型层技术 - 大模型与大语言模型不是一回事
人工智能·语言模型·自然语言处理
居7然39 分钟前
DeepSeek OCR:重新定义AI文档处理的“降本增效”新范式
人工智能·算法·语言模型·自然语言处理·大模型·ocr
while(1){yan}1 小时前
数据结构之堆
数据结构·python·算法
xingxing_F1 小时前
Topaz Video AI for Mac AI视频无损放大 视频画质增强
人工智能·macos·音视频
普蓝机器人1 小时前
面向智慧农业的自主移动果蔬采摘机器人:融合视觉识别与自动驾驶的智能化农作系统研究
人工智能·学习·机器人·移动机器人·三维仿真导航