コード例 #1
0
ファイル: api.py プロジェクト: spankmasters/tensorfuse
 def __call__(self, name=None):
     if name is None:
         name = self.name
     elif ':' in name:
         name = name.split(':')[0]
     return tf_var_from_shape(name, None, self.dtype,
                              len(self.broadcastable))
コード例 #2
0
ファイル: api.py プロジェクト: spankmasters/tensorfuse
def tensor4(name, dtype=None, fixed_shape=None):
    return tf_var_from_shape(name, fixed_shape, dtype, ndim=4)
コード例 #3
0
ファイル: api.py プロジェクト: spankmasters/tensorfuse
def vector(name, dtype=None, fixed_shape=None):
    return tf_var_from_shape(str(name), fixed_shape, dtype, ndim=1)
コード例 #4
0
ファイル: api.py プロジェクト: spankmasters/tensorfuse
def matrix(name, dtype=None, fixed_shape=None):
    return tf_var_from_shape(name, fixed_shape, dtype, ndim=2)
コード例 #5
0
ファイル: api.py プロジェクト: MLDL/tensorfuse
def tensor4(name, dtype=None, fixed_shape=None):
    return tf_var_from_shape(name, fixed_shape, dtype, ndim=4)
コード例 #6
0
ファイル: api.py プロジェクト: MLDL/tensorfuse
def vector(name, dtype=None, fixed_shape=None):
    return tf_var_from_shape(str(name), fixed_shape, dtype, ndim=1)
コード例 #7
0
ファイル: api.py プロジェクト: MLDL/tensorfuse
def matrix(name, dtype=None, fixed_shape=None):
    return tf_var_from_shape(name, fixed_shape, dtype, ndim=2)
コード例 #8
0
ファイル: api.py プロジェクト: MLDL/tensorfuse
 def __call__(self, name=None):
     if name is None:
         name = self.name
     elif ':' in name:
         name = name.split(':')[0]
     return tf_var_from_shape(name, None, self.dtype, len(self.broadcastable))