torch.sparse.softmax¶
- torch.sparse.softmax(input, dim, *, dtype=None) Tensor ¶
应用 softmax 函数。
Softmax 定义为
其中 在稀疏张量索引上运行,未指定的条目将被忽略。这等效于将未指定的条目定义为负无穷大,以便当具有索引的条目未指定时,。
它应用于沿 dim 的所有切片,并将重新缩放它们,以便元素位于 [0, 1] 范围内并总计为 1。
- 参数
input (Tensor) – 输入
dim (int) – 计算 softmax 的维度。
dtype (
torch.dtype
, 可选) – 返回张量的所需数据类型。如果指定,则在执行操作之前将输入张量转换为dtype
。这对于防止数据类型溢出很有用。默认值:None