我正在做一个项目,我想实现ReLU平方激活函数(max {0,x ^ 2})。可以这样称呼它吗: # example code def forward(self, x): s = torch.relu(x**2) return s
確定! 回上一頁