I want to change this Keras function into PyTorch. Should I rewrite a new function or there is a function that Pytorch had already exist? Thanks
batch=input_shape.dim_size(0)
i_h=input_shape.dim_size(1)
i_w=input_shape.dim_size(2)
i_ch=input_shape.dim_size(3)
k_h=filter_shape.dim_size(0)
k_w=filter_shape.dim_size(1)
o_ch=filter_shape.dim_size(3)