@@ -25,7 +25,7 @@ def Visualise(A:Tensor,B:Tensor,C:Tensor,activation:Tensor,points:list[Tensor]|T
2525 res :tuple [int ]= (200 ,200 ), cmaps :list [str ]= [], add_lines_functions :list [FunctionType ]| None = None ,
2626 add_line_args :list [dict ]| None = None ,vmin :int | list [int ]| None = None ,vmax :int | list [int ]| None = None ,
2727 matricies :Tensor | list [Tensor ]| None = None , show :bool = True ,block :bool = True , clr_labels :list [str ]| None = None , depth :int = 2 , link_ax :str | list | None = 'all' ,
28- cursor :bool = False , arangement :tuple | None = None , titles :list [str ]| None = None , call_abs = False , norm_axes = None ) -> None :
28+ cursor :bool = False , arrangement : tuple | None = None , arangement :tuple | None = None , titles :list [str ]| None = None , call_abs = False , norm_axes = None ) -> None :
2929 '''
3030 Visualises any number of fields generated from activation to the plane ABC and arranges them in a (1,N) grid \n
3131 :param A: Position of the top left corner of the image
@@ -79,6 +79,8 @@ def Visualise(A:Tensor,B:Tensor,C:Tensor,activation:Tensor,points:list[Tensor]|T
7979 results = []
8080 lines = []
8181
82+ if arangement is None and arrangement is not None :
83+ arangement = arrangement
8284
8385
8486 if colour_function_args is None and colour_functions is not None :
0 commit comments