引数に＊は必要ないのでしょうか？

Question

引数に＊は必要ないのか疑問に思っています。
http://www.buildinsider.net/small/booktensorflow/0201
を参考にして今Tensorflowの勉強をしています。

def _get_weights(shape,stddev=1.0):
  var = tf.get_variable(
      'weights',
      shape,
      initializer=tf.truncated_normal_initializer(stddev=stddev)
  )
  return var

weights = _get_weights(shape=[5,5,3,64],stddev=1e-4)

というコードが出てきて、_get_weightsメソッドのshapeに*をつけなくていいのか？と思います。
つまり、

def _get_weights(*shape,stddev=1.0):

になると思うのです。
Pythonで、関数で引数にリストを渡すときには*をつけなければならないと習ったので、なぜここではつけないのかな、と。
ちなみに、全体は

# coding: utf-8

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import tensorflow as tf

NUM_CLASSES = 10

def _get_weights(shape,stddev=1.0):
  var = tf.get_variable(
      'weights',
      shape,
      initializer=tf.truncated_normal_initializer(stddev=stddev)
  )
  return var

def _get_biases(shape,value=0.0):
  var = tf.get_variable(
      'biases',
      shape,
      initializer=tf.constant_initializer(value)
  )
  return var

def inference(image_node):
   #  conv1
   with tf.variable_scope('conv1') as scope:
     weights = _get_weights(shape=[5,5,3,64],stddev=1e-4)
     conv = tf.nn.conv2d(image_node,weights,[1,1,1,1],padding='SAME')
     biases = _get_biases([64],value=0.1)
     bias = tf.nn.bias_add(conv,biases)
     conv1 = tf.nn.relu(bias,name=scope.name)

   #  pool
   pool1 = tf.nn.max_pool(conv1,ksize=[1,3,3,1],strides=[1,2,2,1],padding='SAME',name='pool1')

   #  conv2
   with tf.variable_scope('conv2') as scope:
     weights = _get_weights(shape=[5,5,64,64],stddev=1e-4)
     conv = tf.nn.conv2d(pool1,weights,[1,1,1,1],padding='SAME')
     biases = _get_biases([64],value=0.1)
     bias = tf.nn.bias_add(conv,biases)
     conv2 = tf.nn.relu(bias,name=scope.name)

   #  pool2
   pool2 = tf.nn.max_pool(conv2,ksize=[1,3,3,1],strides=[1,2,2,1],padding='SAME',name='pool2')
   reshape = tf.reshape(pool2,[1,-1])
   dim = reshape.get_shape()[1].value

   #  fc3
   with tf.variable_scope('fc3') as scope:
       weights = _get_weights(shape=[dim,384],stddev=0.04)
       biases = _get_biases([384],value=0.1)
       fc3 = tf.nn.relu(
           tf.matmul(reshape,weights) + biases,
           name=scope.name
       )

   #  fc4
   with tf.variable_scope('fc4') as scope:
       weights = _get_weights(shape=[384,192],stddev=0.04)
       biases = _get_biases([192],value=0.1)
       fc4 = tf.nn.relu(tf.matmul(fc3,weights) + biases,name=scope.name)

   #   output
   with tf.variable_scope('output') as scope:
       weights = _get_weights(shape=[192,NUM_CLASSES],stddev=1/192.0)
       biases = _get_biases([NUM_CLASSES],value=0.0)
       logits = tf.add(tf.matmul(fc4,weights),biases,name='logits')


   return logits

のようになっています。

Answer 1

Pythonで、関数で引数にリストを渡すときには*をつけなければならないと習ったので、

教わった内容が間違っているか、誤解して覚えている可能性があります。

リストのまま渡したければ * を付ける必要は無いですし、リストを展開して渡したければ * を付けます。

• 参考: 引数リストのアンパック

Answer 2

*を付けることもできるし、もし付けると、呼び出す側は

weights = _get_weights(5,5,3,64,stddev=1e-4)

こう書くことになります。

Comments

[退会済みユーザー]

なるほど！＊をつけるときは関数の引数にリスト型で持たせるのではなく、一個一個、変数の値でもたせるという認識であっていますか？

[yuba]

はい、こういう呼び方を可変長引数で渡す、といいます。