Как я мог бы присвоить значение с помощью тензорного среза в tensorrt?

#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]