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]])
'''
相关推荐
belldeep7 小时前
python:genesis-world, Genesis-Embodied-AI
人工智能·python·ai·genesis
西索斯coding8 小时前
gemini-3.1-flash-image 报 400 怎么办?图片能打开为何仍是 INVALID_ARGUMENT?
ai
七牛云行业应用8 小时前
DeepSeek Super-agent 与 Grok Bot:开源 Agent 架构和聊天助手有何区别
人工智能·ai·架构·ai编程
梦想不只是梦与想9 小时前
PyTorch、CPU、GPU、CUDA
人工智能·pytorch·python·深度学习·机器学习·cuda
七牛云行业应用10 小时前
MIT “权重空间编程”:GPT-6 Astra 修改 0.016% 的 Qwen 权重,MATH-500 从 7.9% 升至 48.1%
人工智能·gpt·ai
wflynn12 小时前
Meta 816B 参数 Muse Spark Open 在 NVIDIA GB300 上实现每 GPU 50K tokens 每秒
ai·技术
wflynn13 小时前
MIT 团队用 GPT-6 Astra 改写小模型权重,跳过梯度训练提升指令遵循能力
ai·技术
小七-七牛开发者15 小时前
Coding Agent 推理性能优化:Decode、KV Cache 与多 Session 调度
ai·agent·kvcache·ai coding·speculativedecoding
GISer_Jing17 小时前
SuperAgent
ai·langchain
YYYing.17 小时前
【Agent系列 (十一) 】上下文工程
人工智能·python·ai·agent·上下文工程