tensor补充

文章目录

矩阵*运算

python 复制代码
import torch
b = torch.arange(0,5000).unsqueeze(-1)
'''
tensor([[   0],
        [   1],
        [   2],
        ...,
        [4997],
        [4998],
        [4999]])
'''
a = torch.arange(0,128,2)
'''
tensor([  0,   2,   4,   6,   8,  10,  12,  14,  16,  18,  20,  22,  24,  26,
         28,  30,  32,  34,  36,  38,  40,  42,  44,  46,  48,  50,  52,  54,
         56,  58,  60,  62,  64,  66,  68,  70,  72,  74,  76,  78,  80,  82,
         84,  86,  88,  90,  92,  94,  96,  98, 100, 102, 104, 106, 108, 110,
        112, 114, 116, 118, 120, 122, 124, 126])
'''

c = b * a

'''
tensor([[     0,      0,      0,  ...,      0,      0,      0],
        [     0,      2,      4,  ...,    122,    124,    126],
        [     0,      4,      8,  ...,    244,    248,    252],
        ...,
        [     0,   9994,  19988,  ..., 609634, 619628, 629622],
        [     0,   9996,  19992,  ..., 609756, 619752, 629748],
        [     0,   9998,  19996,  ..., 609878, 619876, 629874]])
'''
相关推荐
梦想三三44 分钟前
Flask + PyTorch模型部署实战:从训练权重到API接口完整工程解析(附完整代码)
人工智能·pytorch·python·flask·模型推理·ai 工程化
_冷眸_1 小时前
用 OpenAI SDK 统一调用 Claude、Gemini、DeepSeek:聊聊 AI API 聚合网关的实践
人工智能·ai·llm·agent
白日做梦Q3 小时前
PyTorch从零手写CNN实现NEU钢材表面缺陷检测(完整实战·单篇完结)
人工智能·pytorch·cnn
想会飞的蒲公英3 小时前
TF-IDF + 随机森林中文文本分类全链路实战:从训练脚本到 Flask API + Streamlit 前端
人工智能·pytorch·python·随机森林·分类·flask·tf-idf
欢醉3 小时前
Hermes Agent安装踩坑复盘
ai·hermes
FriendshipT3 小时前
Ultralytics:解读C3模块
人工智能·pytorch·python·深度学习·目标检测
曦尧3 小时前
Taste-Skill:为 AI 生成界面注入“审美基因“的前端技能框架
ai·自动化
wumingxiaoyao16 小时前
从 0 开始学 AI:第 2 课,AI、机器学习、深度学习和大模型是什么关系?
人工智能·深度学习·机器学习·ai·大模型·llm
leonshi18 小时前
Embedchain 是什么?用最简单的话讲清楚一个"AI 知识问答工具框架"
ai·rag
带刺的坐椅19 小时前
Agent Harness 实战指南:构建生产级 AI Agent 的"马具"框架
java·ai·llm·agent·solon-ai