相关疑难解决方法(0)

PyTorch 中的 .flatten() 和 .view(-1) 有什么区别?

双方.flatten().view(-1)压平PyTorch张量。有什么不同?

  1. 是否.flatten()复制张量的数据?
  2. .view(-1)更快吗?
  3. .flatten()没有不工作的情况?

python pytorch

19
推荐指数
2
解决办法
7204
查看次数

如何交换火炬张量中的轴?

我有一个大小为的火炬张量torch.Size([1, 128, 56, 128])

1是通道,128是宽度和高度。56 是图像堆栈。

我怎样才能将其调整为torch.Size([1, 56, 128, 128])

python permutation pytorch

5
推荐指数
1
解决办法
2万
查看次数

标签 统计

python ×2

pytorch ×2

permutation ×1