如何使用torch添加图像通道？

我已经加载了一张包含3个通道的jpg图像，如何添加最后一个通道以生成包含4个通道的图像张量？

假设您有一张包含3个通道的图像，高度<代码>高度和宽度<代码>宽度：

a = torch.Tensor(3, height, width) -- this is your image
b = torch.Tensor(1, height, width) -- the channel you want to add
c = torch.cat(a,b,1)

一个有效的例子：

th> a = torch.Tensor(3,3,3):fill(1)                                                               
th> b = torch.Tensor(1,3,3):fill(0)
th> c = torch.cat(a,b,1)                                                                   
th> c
(1,.,.) = 
  1  1  1
  1  1  1
  1  1  1

(2,.,.) = 
  1  1  1
  1  1  1
  1  1  1

(3,.,.) = 
  1  1  1
  1  1  1
  1  1  1

(4,.,.) = 
  0  0  0
  0  0  0
  0  0  0
[torch.DoubleTensor of size 4x3x3]

---

假设您有一个具有3个通道的图像，高度<代码>高度和宽度<代码>宽度：

a = torch.Tensor(3, height, width) -- this is your image
b = torch.Tensor(1, height, width) -- the channel you want to add
c = torch.cat(a,b,1)

一个有效的例子：

th> a = torch.Tensor(3,3,3):fill(1)                                                               
th> b = torch.Tensor(1,3,3):fill(0)
th> c = torch.cat(a,b,1)                                                                   
th> c
(1,.,.) = 
  1  1  1
  1  1  1
  1  1  1

(2,.,.) = 
  1  1  1
  1  1  1
  1  1  1

(3,.,.) = 
  1  1  1
  1  1  1
  1  1  1

(4,.,.) = 
  0  0  0
  0  0  0
  0  0  0
[torch.DoubleTensor of size 4x3x3]

---

我很高兴它奏效了。你能把答案标记为接受吗（点击勾号）。谢谢我很高兴它奏效了。你能把答案标记为接受吗（点击勾号）。谢谢