使用遞歸函數實現累加:
def sum(nums:Int*):Int={
if(nums.length == 0) 0
else nums.head + sum(nums.tail:_*)
}
結果爲:sum(1,2,3,4,5)
res7: Int = 15
head指的是:取nums的第一個數值,如:1
tail指的是:取nums除第一個數值之外的後面的數值,如:2,3,4,5
def sum(nums:Int*):Int={
if(nums.length == 0) 0
else nums.head + sum(nums.tail:_*)
}
結果爲:sum(1,2,3,4,5)
res7: Int = 15
head指的是:取nums的第一個數值,如:1
tail指的是:取nums除第一個數值之外的後面的數值,如:2,3,4,5
利用torchsummary觀察每一層的情況 1)按照方式 pip install torchsummary 2)