torch用字符串选激活函数

import torch.nn as nn
import torch
a = "ReLU"  #不用带括号
b = getattr(nn,a)
c = b(inplace = False)(torch.tensor(-1))
print(c)

结果是0

参考网址：(36条消息) 关于nn.ReLU函数_JianzhuWang的博客-CSDN博客_nn.relu