torch.cat() python

import torch A = torch.ones(2,3) B = torch.ones(2,3)*2 print('A:',A) print('A:',A.shape) print('----------------------') print('B:',B) print('B:',B.shape) print('-----------------------') C = torch.ca
相關文章
相關標籤/搜索