摘要:
把维度进行压缩A*1*B就会变成A*B也可以使用torch.squeeze去压缩指定维度,如果压缩的指定维度不为1则返回原数组。 import torch import numpy as np x=np.arange(24).reshape((2,1,3,4)) x = torch.tensor(x 阅读全文
posted @ 2022-12-24 11:48
祥瑞哈哈哈
阅读(44)
评论(0)
推荐(0)
摘要:
就是在指定维度前再插入一个新的维度。 import torch import numpy as np x=np.arange(24).reshape((2,3,4)) x = torch.tensor(x) print(x) y=x.permute((2, 0, 1)) print(y.shape) 阅读全文
posted @ 2022-12-24 08:05
祥瑞哈哈哈
阅读(60)
评论(0)
推荐(0)
摘要:
permute进行的是置换。 permute的dim需要参数表示进行置换的维度。 import torch import numpy as np x=np.arange(24).reshape((2,3,4)) x = torch.tensor(x) print(x) y=x.permute((2, 阅读全文
posted @ 2022-12-24 07:44
祥瑞哈哈哈
阅读(201)
评论(0)
推荐(0)

浙公网安备 33010602011771号