kernel_v = [[0, -1, 0],
[0, 0, 0],
[0, 1, 0]]
>>>torch.FloatTensor(kernel_v).unsqueeze(1)
tensor([[[ 0., -1., 0.]],
[[ 0., 0., 0.]],
[[ 0., 1., 0.]]])
>>> torch.FloatTensor(kernel_v).unsqueeze(0)
tensor([[[ 0., -1., 0.],
[ 0., 0., 0.],
[ 0., 1., 0.]]])