Lines Matching refs:features
35 def _npSoftmax(self, features, dim=-1, log=False):
37 dim = len(features.shape) - 1
38 one_only_on_dim = list(features.shape)
40 e = np.exp(features - np.reshape(
42 features, axis=dim), one_only_on_dim))
69 def _testAll(self, features):
70 self._testSoftmax(features, use_gpu=True)
71 self._testSoftmax(features, log=True, use_gpu=True)
75 features = [[1., 1., 1., 1.], [1., 2., 3., 4.]]
85 np_sm = self._npSoftmax(np.array(features))
92 np_lsm = self._npSoftmax(np.array(features), log=True)
106 features = np.array([[1., 1., 1., 1.], [max, 1., 2., 3.]]).astype(type)
108 tf_log_softmax = nn_ops.log_softmax(features)
