File tree Expand file tree Collapse file tree 6 files changed +1
-630
lines changed Expand file tree Collapse file tree 6 files changed +1
-630
lines changed Original file line number Diff line number Diff line change @@ -66,12 +66,9 @@ let () =
6666 let a = TDSL. range_of_shape ~batch_dims: [ 2 ] ~input_dims: [ 3 ] ~output_dims: [ 4 ] () in
6767 let b = TDSL. range_of_shape ~batch_dims: [ 2 ] ~input_dims: [ 4 ] ~output_dims: [ 5 ] () in
6868 let % op a2 = a *+ " b|i->o; b|i->o => b|i->o" a in
69- Tensor. print ~spy: true ~with_code: false ~with_grad: false `Default @@ a;
7069 let ctx = Utils. capture_stdout_logs (fun () -> Train. forward_and_ctx backend ctx a2) in
71- Tensor. print ~here: [% here] ~with_code: false ~with_grad: false `Default @@ a;
7270 let % op c = b *+ " b|h->o; b|i->h => b|i->o" a in
7371 Utils. capture_stdout_logs (fun () -> Train. forward_and_forget backend ctx c);
74- Tensor. print ~here: [% here] ~with_code: false ~with_grad: false `Default @@ a;
7572 (* let%op d = a *+ "a|i->h; b|h->o => ab|i->o" b in Utils.capture_stdout_logs (fun () ->
7673 Train.forward_and_forget backend ctx d); let%op e = a *+ "b|i->h; b|h->o => i->o" b in
7774 Utils.capture_stdout_logs (fun () -> Train.forward_and_forget backend ctx e); let%op f = a *+
Original file line number Diff line number Diff line change @@ -55,9 +55,6 @@ let hello3 () =
5555 let y = TDSL.O. (( + ) ~label: [ " y" ] (hey * zero_to_twenty) zero_to_twenty) in
5656 Train. set_hosted hey.value;
5757 let routine = Train. to_routine (module Backend ) ctx IDX. empty @@ Train. forward y in
58- Tensor. print ~here: [% here] ~with_code: true ~with_grad: false `Inline zero_to_twenty;
59- Tensor. print ~here: [% here] ~with_code: true ~with_grad: false `Default zero_to_twenty;
60- Tensor. print_tree ~with_grad: false ~depth: 9 zero_to_twenty;
6158 Stdio. printf " \n %!" ;
6259 Train. run routine;
6360 Tensor. print ~here: [% here] ~with_code: true ~with_grad: false `Default y;
@@ -117,7 +114,6 @@ let hello5 () =
117114 let hey = TDSL. range_of_shape ~batch_dims: [ 2 ] ~input_dims: [ 3 ] ~output_dims: [ 4 ] () in
118115 let % op ho = hey ++ " ...|1->... => ...|..." in
119116 Train. forward_and_forget backend ctx ho;
120- Tensor. print ~here: [% here] ~with_code: false ~with_grad: false `Default @@ hey;
121117 Tensor. print ~here: [% here] ~with_code: false ~with_grad: false `Default @@ ho
122118
123119let hello6 () =
Original file line number Diff line number Diff line change @@ -193,8 +193,6 @@ let%track2_sexp _Big_matrix (() : unit) : unit =
193193 let zero_to_twenty = TDSL. range 20 in
194194 let % op yd = (hey * zero_to_twenty) + zero_to_twenty in
195195 Train. forward_and_forget backend ctx yd;
196- Tensor. print ~here: [% here] ~with_code: false ~with_grad: false `Inline zero_to_twenty;
197- Tensor. print ~here: [% here] ~with_code: false ~with_grad: false `Default zero_to_twenty;
198196 Tensor. print ~here: [% here] ~with_code: false ~with_grad: false `Default hey;
199197 Tensor. print ~here: [% here] ~with_code: false ~with_grad: false `Default yd
200198
@@ -215,7 +213,6 @@ let%track2_sexp _Very_big_tensor (() : unit) : unit =
215213 let hey = TDSL. range_of_shape ~batch_dims: [ 6 ] ~input_dims: [ 7 ; 8 ] ~output_dims: [ 9 ] () in
216214 let % op ye = (hey * (1 + 1 )) - 10 in
217215 Train. forward_and_forget backend ctx ye;
218- Tensor. print ~here: [% here] ~with_code: false ~with_grad: false `Default hey;
219216 Tensor. print ~here: [% here] ~with_code: false ~with_grad: false `Default ye
220217
221218let _suspended (() : unit ) : unit =
You can’t perform that action at this time.
0 commit comments