How to “cut” a tensor into half in Pytorch?

前端 未结 0 401
清酒与你
清酒与你 2020-11-27 14:59

I have a tensor with shape [1, 2, 96, 96] and would like two tensors with the shape [1, 1, 96, 96], is there a quick way of doing this? Thanks in advance

相关标签:
回答
  • 消灭零回复
提交回复
热议问题