train-kit / dist / forward.js
1
2
3
4
5
6
import { $APP, shadow$provide } from "./shared.js";
const shadow_esm_import = function(x) { return import(x) };
$APP.ardegazu.train.forward={};$APP.ardegazu.train.forward.mlp_forward=function(a,b){a=a.layers;for(var c=a.length,d=0;;)if(d<c){var e=a[d],f=e.w;e=e.b;for(var g=e.length,h=b.length,k=Array(g),l=d===c-1,m=0;;)if(m<g){var n=m*h;a:{for(var p=0,q=e[m];;)if(p<h)q+=f[n+p]*b[p],p+=1;else{n=q;break a}n=void 0}k[m]=l?n:Math.max(0,n);m+=1}else break;b=k;d+=1}else return b};
$APP.ardegazu.train.forward.mlp_init=function(a){switch(arguments.length){case 1:return $APP.ardegazu.train.forward.mlp_init.cljs$core$IFn$_invoke$arity$1(arguments[0]);case 2:return $APP.ardegazu.train.forward.mlp_init.cljs$core$IFn$_invoke$arity$2(arguments[0],arguments[1]);default:throw Error(["Invalid arity: ",arguments.length].join(""));}};
$APP.ardegazu.train.forward.mlp_init.cljs$core$IFn$_invoke$arity$1=function(a){return $APP.ardegazu.train.forward.mlp_init.cljs$core$IFn$_invoke$arity$2(a,Math.random)};
$APP.ardegazu.train.forward.mlp_init.cljs$core$IFn$_invoke$arity$2=function(a,b){for(var c=[],d=a.length,e=1;;)if(e<d){var f=a[e-1],g=a[e],h=Math.sqrt(2/f),k=Array(g*f),l=Array(g);f*=g;for(var m=0;;)if(m<f)k[m]=((b.cljs$core$IFn$_invoke$arity$0?b.cljs$core$IFn$_invoke$arity$0():b.call(null))*2-1)*h,m+=1;else break;for(h=0;;)if(h<g)l[h]=0,h+=1;else break;c.push({w:k,b:l});e+=1}else break;return{sizes:a.slice(),layers:c}};$APP.ardegazu.train.forward.mlp_init.cljs$lang$maxFixedArity=2;export const mlpForward=$APP.ardegazu.train.forward.mlp_forward;export const mlpInit=$APP.ardegazu.train.forward.mlp_init;

static mirror of HEAD · about · clone: git clone https://git.ardegazu.ro/train-kit.git