最近，NLP明星公司Hugging Face发布了一个叫做Infinity的产品，可以以1ms延时完成Transformer的推理，性能相当高了。

矩阵（Matrix）是一个按照长方阵列排列的复数或实数集合，元素是实数的矩阵称为实矩阵，元素是复数的矩阵称为复矩阵。而行数与列数都等于n的矩阵称为n阶矩阵或n阶方阵。...

故事的起因在于我们在实际工作中遇到的一个小的需求，即我们在模型定义当中需要用到argmax的信息，因此，我们就快速地写下了如下一段代码：

注：本文转载知乎讨论(原始链接见阅读原文)，下面所述观点并不代表本公众号立场，希望给即将进入高校工作的青年科研人员提供一些参考意见，欢迎大家文末留言讨论。...

本文主要向大家分享一个小编刚刚学习的神经网络应用的实例：风格迁移(Neural-Transfer)。这是一个由 Leon A. Gatys，Alexander S. Ecker和Matthias Bethge提出的算法。通过这个算法，我们可以用一种新的风格对指定图片进行...

The reduced tensor, of the same dtype as the input_tensor.

概念： 几何代数中定义的张量是基于向量和矩阵的推广，通俗一点来理解的话，而我们可以把标量视为零阶张量，矢量视为一阶张量，那么矩阵就是二阶张量。...

https://pytorch.org/docs/stable/generated/torch.max.html#torch.max

大家好，欢迎来到我们的星球知识小卡片专栏，本期给大家分享模型压缩中的一个非常重要的方向，动态模型。

torch.chunk(input, chunks, dim = 0) 函数会将输入张量（input）沿着指定维度（dim）均匀的分割成特定数量的张量块（chunks），并返回元素为张量块的元组。torch.chunk 函数有三个参数：...