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]