pytorch 分割一个tensor并求平均

pytorch的torch.add()及torch.split()函数import torch# outputs是一个[batch, seq, 40]维的tensor，把outputs分割成两个[batch, seq, 20]的tensor，并每个元素求平均值add = torch.add(*torch.split(outputs, 20, dim=2)) / 2

林子要加油

4197人浏览 · 2020-05-27 10:05:17

林子要加油 · 2020-05-27 10:05:17 发布

pytorch的torch.add()及torch.split()函数

import torch
# outputs是一个[batch, seq, 40]维的tensor，把outputs分割成两个[batch, seq, 20]的tensor，并每个元素求平均值
add = torch.add(*torch.split(outputs, 20, dim=2)) / 2

AI编程社区

汇聚全球AI编程工具，助力开发者即刻编程。

更多推荐

AI编程工具能力对比评测：构建安全可靠应用的最佳选择

AI编程社区

AI原生应用领域混合推理技术的最新进展

当我们谈论AI原生应用（如ChatGPT、GitHub Copilot、智能诊疗系统）时，“灵活但不可靠”“准确但不智能”的矛盾始终是绕不开的痛点——纯神经模型（如大语言模型）像“凭直觉做题的学生”，擅长处理复杂场景却常犯“幻觉”错误；纯符号系统（如传统专家系统）像“死记硬背的书呆子”，逻辑严谨却无法应对非结构化数据。混合推理技术。