2023年6月7日
摘要: ## einsum求矩阵运算 #### einsum是什么 使用chatGPT解读官方的[文档](https://pytorch.org/docs/stable/generated/torch.einsum.html): ``` Einsum允许使用基于Einstein求和约定的简写格式来计算许多常 阅读全文
posted @ 2023-06-07 15:48 蔚蓝色の天空 阅读(38) 评论(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 蔚蓝色の天空 阅读(32) 评论(0) 推荐(0) 编辑