Pytorch - 使用火炬功能选择张量区域

Question

我正在寻找一种使用火炬函数（不使用 numpy）选择 PyTorch 张量区域的方法。 您对如何进行有任何建议吗？

换句话说，我正在寻找一种裁剪矩阵区域的方法。 使用 numpy，它会是这样的

import numpy as np
A = np.random.rand(16,16)
B = A[0:8, 0:8]

我正在尝试的方法如下：

from torchvision import transforms
A = torch.randn([1,3,64,64])
B = torch.split(A, [16,32,16], dim =2)
C = torch.split(B, [16,32,16], dim =3)

这给出了错误

'tuple' 对象没有属性 'split'

Answer 1

常规切片有什么问题？

import torch

A = torch.randn([1,3,64,64])
B = A[..., 16:32, 16:32]