快捷方式

torch.sparse.softmax

torch.sparse.softmax(input, dim, *, dtype=None) Tensor

应用 softmax 函数。

Softmax 定义为

Softmax(xi)=exp(xi)jexp(xj)\text{Softmax}(x_{i}) = \frac{exp(x_i)}{\sum_j exp(x_j)}

其中 i,ji, j 在稀疏张量索引上运行,未指定的条目将被忽略。这等效于将未指定的条目定义为负无穷大,以便当具有索引的条目未指定时,exp(xk)=0exp(x_k) = 0

它应用于沿 dim 的所有切片,并将重新缩放它们,以便元素位于 [0, 1] 范围内并总计为 1。

参数
  • input (Tensor) – 输入

  • dim (int) – 计算 softmax 的维度。

  • dtype (torch.dtype, 可选) – 返回张量的所需数据类型。如果指定,则在执行操作之前将输入张量转换为 dtype。这对于防止数据类型溢出很有用。默认值:None

文档

访问 PyTorch 的全面开发人员文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源