torch.nn.ReLU用法

torch.nn.ReLU:调用relu函数代码示例来自pytorch官方文档，正好复习下unsqueeze和cat函数的使用：import torchimport torch.nn as nnm=nn.ReLU()input=torch.randn(2).unsqueeze(0)print("input:",input)print("input的shape:",input.shape)outpu

江南汪

10300人浏览 · 2021-05-11 12:35:33

江南汪 · 2021-05-11 12:35:33 发布

torch.nn.ReLU:调用relu函数
代码示例来自pytorch官方文档，正好复习下unsqueeze和cat函数的使用：

import torch
import torch.nn as nn
m=nn.ReLU()
input=torch.randn(2).unsqueeze(0)
print("input:",input)
print("input的shape:",input.shape)
output=torch.cat((m(input),m(-input)),0)
print("output:",output)
print("output的shape:",output.shape)

输出：

input: tensor([[-1.0105,  1.2734]])
input的shape: torch.Size([1, 2])
output: tensor([[0.0000, 1.2734],
        [1.0105, 0.0000]])
output的shape: torch.Size([2, 2])