В pytorch, как я могу сложить некоторые элементы и получить тензор меньшей формы? - PullRequest
Новая
       13

В pytorch, как я могу сложить некоторые элементы и получить тензор меньшей формы?

2 голосов
/ 20 марта 2020

В частности, у меня есть тензор размерности 298x160x160 (грани в 298 кадрах), мне нужно сложить каждый элемент 4x4 в двух последних димесионах, чтобы я мог получить тензор 298x40x40.

Как мне этого добиться?

1 Ответ

1 голос
/ 20 марта 2020

Вы можете создать сверточный слой с одним каналом 4x4 и установить его веса 1, с шагом 4 ( также см. Conv2D do c): 

a = torch.ones((298,160,160))
# add a dimension for the channels. Conv2D expects the input to be : (N,C,H,W)
# where N=number of samples, C=number of channels, H=height, W=width
a = a.unsqueeze(1)
a.shape

Out: torch.Size([298, 1, 160, 160])

with torch.no_grad(): # I assume you don't need to backprop, otherwise remove this check
    m = torch.nn.Conv2d(in_channels=1, out_channels=1, kernel_size=4,stride=4,bias=False)
    # set the kernel values to 1
    m.weight.data = m.weight.data * 0. + 1.
# apply the kernel and squeeze the channel dim out again
res = m(a).squeeze()
res.shape

Out: torch.Size([298, 40, 40])

...