torch.sparse.log_softmax

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

先应用softmax函数，然后计算结果的对数。

更多详情请参见softmax。

参数

• 输入 (Tensor) – 输入张量

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

• dtype (torch.dtype, 可选) – 返回张量的数据类型。如果指定了 dtype，则在执行操作前将输入张量转换为指定的 dtype，以防止数据溢出。默认值：None