Lines Matching refs:features
36 def _npXent(self, features, labels, dim=-1):
38 dim = len(features.shape) - 1
39 one_only_on_dim = list(features.shape)
42 features - np.reshape(np.amax(features, axis=dim), one_only_on_dim))
67 def _testAll(self, features, labels):
68 self._testXent(features, labels, use_gpu=False)
69 self._testXent(features, labels, use_gpu=True)
99 features = [[1., 1., 1., 1.], [1., 2., 3., 4.]]
121 np_loss, np_backprop = self._npXent(np.array(features), np.array(labels))
237 features = np.array([[[1., 1., 1., 1.], [1., 2., 3., 4.]],
245 self._testXentWrapper(features, labels, dim=0, use_gpu=False)
246 self._testXentWrapper(features, labels, dim=0, use_gpu=True)
247 self._testXentWrapper(features, labels, dim=1, use_gpu=False)
248 self._testXentWrapper(features, labels, dim=1, use_gpu=True)
249 self._testXentWrapper(features, labels, dim=-1, use_gpu=False)
250 self._testXentWrapper(features, labels, dim=-1, use_gpu=True)
253 features = np.zeros([0, 2, 4]).astype(np.float32)
255 np_loss, _ = self._npXent(features, labels)
258 labels=labels, logits=features)
