Pytorch - выбрать область тензора с помощью функции torch

Question

Я ищу способ выбора области тензора PyTorch, используя функцию факела (без использования numpy). У вас есть предложения, как поступить?

Другими словами, я ищу способ обрезать область матрицы. Используя 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]