package Ġw um o . sim . tensorflow . ops Ċ Ċ import Ġorg . junit . Test Ċ import Ġw um o . sim . tensorflow . tf Ċ import Ġw um o . sim . tensorflow . types . INT 3 2 Ċ import Ġw um o . sim . util . Shape Ċ import Ġw um o . sim . util . f Ċ Ċ class ĠMath _ ops Kt Test Ġ: ĠBaseTest () Ġ{ ĊĠĠĊĠ Ġ@ Test ĊĠ Ġfun Ġmat mul () Ġ{ ĊĠĠĠ Ġval ĠA Ġ= Ġtf . const ( Shape ( 1 , Ġ 4 ), Ġf ( 1 f , Ġ 2 f , Ġ 3 f , Ġ 4 f )) ĊĠĠĠ Ġval ĠB Ġ= Ġtf . const ( Shape ( 4 , Ġ 1 ), Ġf ( 1 f , Ġ 2 f , Ġ 3 f , Ġ 4 f )) ĊĠĠĠ Ġval ĠC Ġ= Ġtf . mat Mul ( A , ĠB ) ĊĠĠĠĠĊĠĠĠ Ġprint Graph () ĊĠĠĠ Ġtf . session Ġ{ ĊĠĠĠĠĠ ĠC . eval () ĊĠĠĠ Ġ} ĊĠ Ġ} ĊĠĠĊĠ Ġ@ Test ĊĠ Ġfun Ġadd N () Ġ{ ĊĠĠĠ Ġval Ġa Ġ= Ġtf . const ( Shape ( 2 , Ġ 2 ), Ġ 1 f , Ġname Ġ= Ġ" a ") ĊĠĠĠ Ġval Ġb Ġ= Ġtf . const ( Shape ( 2 , Ġ 2 ), Ġ 2 f , Ġname Ġ= Ġ" b ") ĊĠĠĠ Ġval Ġc Ġ= Ġtf . const ( Shape ( 2 , Ġ 2 ), Ġ 3 f , Ġname Ġ= Ġ" c ") ĊĠĠĠ Ġval Ġd Ġ= Ġtf . add N ( listOf ( a , Ġb , Ġc ), Ġname Ġ= Ġ" add n ") ĊĠĠĠ Ġprint Graph () ĊĠĠĠ Ġtf . session Ġ{ ĊĠĠĠĠĠ Ġd . eval () ĊĠĠĠ Ġ} ĊĠ Ġ} ĊĠĠĊĠ Ġ@ Test ĊĠ Ġfun Ġcast () Ġ{ ĊĠĠĠ Ġval Ġa Ġ= Ġtf . const ( Shape ( 2 , Ġ 2 ), Ġ 2 . 5 f , Ġ" a ") ĊĠĠĠ Ġval Ġb Ġ= Ġtf . cast ( a , ĠINT 3 2 , Ġ" b ") ĊĠĠĠ Ġprint Graph () ĊĠĠĠ Ġtf . session Ġ{ ĊĠĠĠĠĠ Ġb . eval () ĊĠĠĠ Ġ} ĊĠ Ġ} ĊĠĠĊĠ Ġ@ Test ĊĠ Ġfun Ġarg max () Ġ{ ĊĠĠĠ Ġval ĠA Ġ= Ġtf . const ( Shape ( 3 , Ġ 3 ), Ġ 1 f ) ĊĠĠĠ Ġval ĠB Ġ= Ġtf . argmax ( A , Ġ 1 ) ĊĠĠĠ Ġtf . session Ġ{ ĊĠĠĠĠĠ ĠB . eval () ĊĠĠĠ Ġ} ĊĠ Ġ} Ċ }