Pytorch 不同通道分别激活

截取出来不同的通道，进行激活即可

demo:

import torch
import torch.nn as nn
import numpy as np

a = torch.randn((1, 2, 1, 1))

relu = nn.LeakyReLU()

b = relu(a)
c = relu(a[:, 0])
d = relu(a[:, 1])

print('a:\n', a)
print('b:\n', b)
print('c:\n', c)
print('d:\n', d)

# ANTenna 2020.07.16