Home
last modified time | relevance | path

Searched refs:swapaxes (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/nn/metrics/
Daccuracy.py100 y_pred = y_pred.swapaxes(1, dimension_index).reshape(-1, self._class_num)
101 y = y.swapaxes(1, dimension_index).reshape(-1, self._class_num)
Dprecision.py117 y_pred = y_pred.swapaxes(1, 0).reshape(class_num, -1)
118 y = y.swapaxes(1, 0).reshape(class_num, -1)
Drecall.py117 y_pred = y_pred.swapaxes(1, 0).reshape(class_num, -1)
118 y = y.swapaxes(1, 0).reshape(class_num, -1)
/third_party/mindspore/tests/st/numpy_native/
Dtest_array_ops.py254 a = mnp.swapaxes(input_tensor, 0, 1)
255 b = mnp.swapaxes(input_tensor, 1, 0)
256 c = mnp.swapaxes(input_tensor, 1, 1)
257 d = mnp.swapaxes(input_tensor, 2, 1)
258 e = mnp.swapaxes(input_tensor, 1, 2)
259 f = mnp.swapaxes(input_tensor, 2, 2)
264 a = onp.swapaxes(input_array, 0, 1)
265 b = onp.swapaxes(input_array, 1, 0)
266 c = onp.swapaxes(input_array, 1, 1)
267 d = onp.swapaxes(input_array, 2, 1)
[all …]
/third_party/mindspore/mindspore/numpy/
D__init__.py29 from .array_ops import (transpose, expand_dims, squeeze, rollaxis, swapaxes, reshape,
Darray_ops.py224 def swapaxes(x, axis1, axis2): function
251 return x.swapaxes(axis1, axis2)
/third_party/mindspore/mindspore/_extends/parse/
Dstandard_method.py356 def swapaxes(x, axis1, axis2): function
/third_party/mindspore/mindspore/nn/layer/
Dthor_layer.py440 self.weight_init = Tensor(weight_init.asnumpy().swapaxes(0, 1), weight_init.dtype)
/third_party/mindspore/mindspore/common/
Dtensor.py702 def swapaxes(self, axis1, axis2): member in Tensor