可以给我举一个使用torch.softmax()的例子吗

当你需要将一组数转换为概率分布时，可以使用torch.softmax()函数。

例如，假设我们有一个包含三个元素的张量tensor=[1,2,3]，我们想将其转换为概率分布，可以使用以下代码：

import torch
tensor = torch.tensor([1,2,3])
prob_dist = torch.softmax(tensor, dim=0)
print(prob_dist)

输出结果为：

tensor([0.0900, 0.2447, 0.6652])

这意味着第一个元素的概率为0.09，第二个元素的概率为0.2447，第三个元素的概率为0.6652，且三个概率之和为1。