合并:
1
2
3
|
torch.cat(inputs = (a, b), dimension = 1 ) e.g. x = torch.cat((x,y), 0 ) 沿x轴合并 |
截取:
1
|
x[:, 2 : 4 ] |
以上这篇pytorch中tensor的合并与截取方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持服务器之家。
原文链接:https://blog.csdn.net/znzhizs/article/details/79616735