import mindspore as ms
import mindspore.common.initializer as init
import mindspore.nn as nn

in_t = Tensor(shape = (1,256,32,32),dtype = ms.float32.init = init.Normal())

conv = nn.Conv2d(256,256,3,2,padding = 3//2,pad_mode = "pad",has_bias = True)
conv_t = nn.Conv2dTranspose(256,256,3,padding = 1,,stride = 2,pad_mode = "pad",output_padding = 1,has_bias = True)
conv_tt = nn.Conv2dTranspose(256,256,3,padding = 1,,stride = 2,pad_mode = "pad",output_padding = 1)

out1 = conv(in_t)
out2 = conv_t(out1)
out3 = conv_tt(out1)
print(out1.shape,out.shape,out.shape)