本文整理汇总了Python中theano.tensor.argmin方法的典型用法代码示例。如果您正苦于以下问题:Python tensor.argmin方法的具体用法?Python tensor.argmin怎么用?Python tensor.argmin使用的例子?那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类theano.tensor
的用法示例。
在下文中一共展示了tensor.argmin方法的4个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: argmin
# 需要导入模块: from theano import tensor [as 别名]
# 或者: from theano.tensor import argmin [as 别名]
def argmin(x, axis=-1):
return T.argmin(x, axis=axis, keepdims=False)
示例2: get_output_for
# 需要导入模块: from theano import tensor [as 别名]
# 或者: from theano.tensor import argmin [as 别名]
def get_output_for(self, inputs):
A = inputs[0]
X = inputs[1]
max_degree_node = T.argmax(A.sum(0))
min_degree_node = T.argmin(A.sum(0))
return self.reduce(A, [max_degree_node, min_degree_node])
示例3: _get_cluster_symbol
# 需要导入模块: from theano import tensor [as 别名]
# 或者: from theano.tensor import argmin [as 别名]
def _get_cluster_symbol(self):
output = self._get_output_symbol()
Y_hat = T.reshape(output, (self.batch, self.y_n, self.k))
y = self._get_y_symbol()
Y = T.tile(y[:, :, None], (1, 1, self.k))
diff = T.mean((Y - Y_hat)**2, axis=1)
cluster = T.argmin(diff, axis=1)
return cluster
示例4: get_output_for
# 需要导入模块: from theano import tensor [as 别名]
# 或者: from theano.tensor import argmin [as 别名]
def get_output_for(self, input, **kwargs):
if self.mode in ('max', 'min'):
reductor = T.argmax if self.mode == 'max' else T.argmin
indices1 = T.arange(input.shape[0])
indices2 = reductor(input[:, :, 0], axis=1)
return input[indices1, indices2]
elif self.mode == 'mean':
return T.mean(input, axis=1)