package Ġeu . red bean . kon nan . layers Ċ Ċ import Ġeu . red bean . k ten . api . auto grad . utils . reshape Ċ import Ġeu . red bean . k ten . api . tensor . Tensor Ċ Ċ class ĠRes hape ( ĊĠĠĠ Ġvar arg Ġval Ġnew Shape : ĠInt , ĊĠĠĠ Ġname : ĠString ? Ġ= Ġnull Ċ ): ĠLayer ( name ) Ġ{ ĊĊĠĠĠ Ġoverride Ġfun Ġpost Invoke () Ġ{ ĊĠĠĠĠĠĠĠ Ġshape Ġ= Ġprevious Layers [ 0 ]. shape . reshape ( new Shape . toList ()) ĊĠĠĠ Ġ} ĊĊĠĠĠ Ġoverride Ġfun Ġforward ( input : ĠTensor ): ĠTensor Ġ{ ĊĠĠĠĠĠĠĠ Ġreturn Ġinput . reshape (- 1 , Ġ* shape . to IntArray ()) Ġ// TODO Ġchange Ġit Ġwhen Ġtime Ġdist Ġlayer Ġis Ġimplemented ĊĠĠĠ Ġ} Ċ Ċ }