摘要:
## einsum求矩阵运算 #### einsum是什么 使用chatGPT解读官方的[文档](https://pytorch.org/docs/stable/generated/torch.einsum.html): ``` Einsum允许使用基于Einstein求和约定的简写格式来计算许多常 阅读全文
posted @ 2023-06-07 15:48
蔚蓝色の天空
阅读(77)
评论(0)
推荐(0)
摘要:
#### einops方法 该方法可以快速实现矩阵的快速变化。 ``` import torch import torch.nn as nn from einops import rearrange # 快速矩阵变化 class TestAttentionQKV: def __init__(self 阅读全文
posted @ 2023-06-07 15:42
蔚蓝色の天空
阅读(94)
评论(0)
推荐(0)
浙公网安备 33010602011771号