#pytorch #tensorrt
Вопрос:
Я хочу сделать что-то вроде кода ниже:
out_h = (H 2*padding[0] - ksize[0])//stride[0] 1 out_w = (W 2*padding[1] - ksize[1])//stride[1] 1 y_max = y stride[0] * out_h x_max = x stride[1] * out_w out[:, :, y, x, :, :] = img[:, :, y:y_max:stride[0], x:x_max:stride[1]] Правую сторону можно сделать вот так, как я мог бы ее назначить out[:, :, y, x, :, :] ?
img_slice = ctx.network.add_slice( input=img , start=[0,0,y,x] , shape=[N,C,out_h,out_w] , stride=[stride[0], stride[1]] ).get_output[0]