自定义博客皮肤VIP专享

*博客头图：

点击选择上传的图片

格式为PNG、JPG，宽度*高度大于1920*100像素，不超过2MB，主视觉建议放在右侧，请参照线上博客头图

请上传大于1920*100像素的图片！

博客底图：

点击选择上传的图片

图片格式为PNG、JPG，不超过1MB，可上下左右平铺至整个背景

栏目图：

点击选择上传的图片

图片格式为PNG、JPG，图片宽度*高度为300*38像素，不超过0.5MB

主标题颜色：

RGB颜色，例如：#AFAFAF

Hover：

RGB颜色，例如：#AFAFAF

副标题颜色：

RGB颜色，例如：#AFAFAF

预览取消提交

自定义博客皮肤

上一步保存

weixin_45704823的博客

原创 Multi-Head Self-Attention

在示例的最后部分，我们创建一个多头自注意力层并将其应用于输入张量。

2023-04-14 13:20:01 417 1

原创 Categorical类

类是PyTorch中的一个类，用于处理离散分类概率分布。它可以处理概率分布（probs）或者logits（对数概率）。# 假设我们有一个离散概率分布，有3个类别，概率分别为[0.1, 0.5, 0.4]类提供了一些实用方法，如生成随机样本、计算概率或对数概率等。# 使用概率创建Categorical对象。# 从分布中采样一个随机动作。# 计算给定动作的对数概率。# 计算给定动作的概率。

2023-04-10 11:47:10 1290 1