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]])
'''
相关推荐
Zguigo5 小时前
【CUDA1】GPUvsCPU,CUDA Kernel
人工智能·pytorch·深度学习
三声三视9 小时前
75 条文章索引被一条 add 清成 1 条,退出码还是 0:tri-article 的 index.py 我读了 205 行
人工智能·ai·skill·tri-skills·tri-article
牛油果子哥q9 小时前
生产级AI项目上线全流程:Docker容器化、服务编排、监控告警、日志收集、容灾降级、线上运维闭环
人工智能·ai
星辰徐哥9 小时前
本地视频预览别只自己看:把Remotion动效项目发给客户远程验收
docker·ai·node.js·html·音视频·react·remotion
HYDtomako12 小时前
Agent checkpoint设计
ai·agent·checkpoint·claude code
七灵微13 小时前
【AI】代码实战- 基于时间序列的轴承检测1
人工智能·机器学习·ai·pandas·matplotlib
RSABLOCKCHAIN13 小时前
读懂AI-Agent项目Career-ops项目的底层原理
ai
十正14 小时前
把表变成模型够得着的句柄
人工智能·ai·agent
Jared_devin14 小时前
单头、多头 Self-Attention可视化
人工智能·pytorch·深度学习·算法·机器学习·pycharm
长谷深风11115 小时前
支付审批的智能风险控制设计
大数据·人工智能·ai·大模型·支付·aiagent·hitl