einops

import einops # 创建一个形状为(batch_size, seq_length, hidden_dim)的张量 tensor = tf.constant([[[1, 2, 3, 4], [5, 6, 7, 8]], [[9, 10, 11, 12], [13, 14, 15, 16]] ......

参考资料： https://einops.rocks/1-einops-basics/ 这篇随笔主要记录使用Einops这个矩阵操作的库的时候发现的，值得记录的高频或者重要的操作： 既然网上的教程已经很详尽了，为什么还要这样做？因为遇见一次的情况很有可能下次还会遇到，主要是写给自己看，方便以后查阅。 ......

#### einops方法 该方法可以快速实现矩阵的快速变化。 ``` import torch import torch.nn as nn from einops import rearrange # 快速矩阵变化 class TestAttentionQKV: def __init__(self ......