Loading

2021年3月30日
torch小知识
摘要： 1.torch.eq out = torch.eq(input, other) 功能： 比较两个张量的元素是否相同 参数： input：输入的张量 other: 用于比较的张量 out: 输出元素为True或者是Flase的张量 例子： 例子1： >>> torch.eq(torch.tensor( 阅读全文
posted @ 2021-03-30 22:26 Guang'Jun 阅读(160) 评论(0) 推荐(0)
在torch中softmax的使用
摘要： ## 在torch中softmax的使用在torch中softmax的使用 在哪一维度上进行softmax操作，哪一维度的值之和为1 class Fc(nn.Module): def __init__(self, in_channel, out_channel): super(Fc, self)._ 阅读全文
posted @ 2021-03-30 14:29 Guang'Jun 阅读(2692) 评论(0) 推荐(0)