在Chapel中使用矩阵创建域

时间:2017-08-26 20:08:54

标签: matrix chapel

我有一个域D,我想用它来索引几个矩阵A。形式的东西

var dom: domain(1) = {0..5};
var mats: [dom] <?>;

var a0 = [[0.0, 0.1, 0.2], [0.3, 0.4, 0.5]];
var a1 = [[1.0, 1.1, 1.2, 1.3], [1.4, 1.5, 1.6, 1.7]];

mats[0] = a0;
mats[1] = a1;

每个a都是2D,但大小不同。是的,其中一些将是稀疏的(但不一定是出于这个问题的目的)

== UPDATE ==

为清楚起见,我有一系列图层(它是一个神经网络),比如1..15。我创建了var layerDom = {1..15}每个图层都有多个与之关联的对象,比如错误,所以我有

var errors: [layerDom] real;  // Just a number

我想要

var Ws: [layerDom] <matrixy thingy>;  // Weight matrices all of different shape.

1 个答案:

答案 0 :(得分:2)

从Chapel 1.15开始,没有一种优雅的方法可以创建一个阵列数组,其中内部数组具有不同的大小。这是因为内部数组都共享同一个域,这意味着更改一个数组的域会更改所有数组。

要获得所需的效果,您需要创建一个包含数组的记录/类数组:

record Weight {
  var D : domain(2);
  var A : [D] real;
}

var layers = 4;
var weights : [1..layers] Weight;
for i in 1..layers {
  weights[i].D = {1..i, 1..i};
  weights[i].A = i;
}

for w in weights do writeln(w.A, "\n");

// 1.0
// 
// 2.0 2.0
// 2.0 2.0
// 
// 3.0 3.0 3.0
// 3.0 3.0 3.0
// 3.0 3.0 3.0
// 
// 4.0 4.0 4.0 4.0
// 4.0 4.0 4.0 4.0
// 4.0 4.0 4.0 4.0
// 4.0 4.0 4.0 4.0
//