generateTensor3D function

Tensor<Tensor3D> generateTensor3D(
  1. int depth,
  2. int rows,
  3. int cols
)

Implementation

Tensor<Tensor3D> generateTensor3D(int depth, int rows, int cols) {
  Random random = Random();
  Tensor3D t3d = [];
  for (int d = 0; d < depth; d = d + 1) {
    Matrix m = [];
    for (int i = 0; i < rows; i = i + 1) {
      Vector row = [];
      for (int j = 0; j < cols; j = j + 1) {
        row.add(random.nextDouble());
      }
      m.add(row);
    }
    t3d.add(m);
  }
  return Tensor<Tensor3D>(t3d);
}