blob: 1389ad6c40593831058b8383e828229c31bf1bc4 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
object Test {
class Range(from: Int, end: Int) {
inline
def foreach(inline op: Int => Unit): Unit = {
var i = from
while (i < end) {
op(i)
i += 1
}
}
def filter(p: Int => Boolean): List[Int] = ???
}
implicit class intWrapper(private val start: Int) extends AnyVal {
def until(end: Int) = new Range(start, end)
def to(limit: Int) = new Range(start, limit + 1)
}
def matmul(xs: Array[Array[Double]], ys: Array[Array[Double]]): Array[Array[Double]] = {
def nrows = xs.length
def ncols = ys(0).length
def n = ys.length
assert(xs(0).length == n)
val zs = Array.ofDim[Double](nrows, ncols)
for (i <- intWrapper(0) until nrows)
for (j <- 0 until ncols) {
var x = 0.0
for (k <- 0 until n)
x += xs(i)(k) * ys(k)(j)
zs(i)(j) = x
}
zs
}
def main(args: Array[String]) = {
1.until(10).foreach(i => println(i))
1.until(10).foreach(println(_))
1.until(10).foreach(println)
for (i <- 1 to 10) println(i)
for (k1 <- 1 to 10)
for (k2 <- 1 to 10)
println(s"$k1")
}
}
|