IVD-Net01:代码涉及函数
1.torch.cat
将两个张量(tensor)拼接在一起,cat是concatnate的意思,即拼接,联系在一起。
A=torch.ones(2,3) #2x3的张量(矩阵)
B=torch.ones(4,3)#4x3的张量(矩阵)
C=torch.cat((A,B),0)#按维数0(行)拼接
C.size() = [6, 3]
2.Dataloader(object)的参数:
dataset(Dataset): 传入的数据集
batch_size(int, optional):
原创
2021-03-11 22:19:35 ·
142 阅读 ·
0 评论