torch.sparse.log_softmax

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

应用softmax函数后跟对数。

查看 softmax 了解更多详情。

Parameters

• 输入 (张量) – 输入

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

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