mindspore.ops.rot90
- mindspore.ops.rot90(input, k, dims)[源代码]
沿指定维度的平面内将n-D tensor旋转90度。 如果k>0,旋转方向是从第一轴朝向第二轴,如果k<0,旋转方向从第二轴朝向第一轴。
警告
从2.9.0(不含)之后版本开始,参数 k 和 dims 将补充默认值,接口签名将变更为
mindspore.ops.rot90(input, k=1, dims=(0, 1))。 该变更与当前版本兼容,显式传入 k 和 dims 时行为不变。- 参数:
input (Tensor) - 输入tensor。
k (int) - 旋转的次数。2.9.0(不含)之后版本默认值将变更为
1。dims (Union[list(int), tuple(int)]) - 指定维度。2.9.0(不含)之后版本默认值将变更为
(0, 1)。
- 返回:
Tensor
- 支持平台:
AscendGPUCPU
样例:
>>> import mindspore >>> x = mindspore.tensor([[1, 2, 3], ... [4, 5, 6]], mindspore.float32) >>> output = mindspore.ops.rot90(x, k=1, dims=(0, 1)) >>> print(output) [[3. 6.] [2. 5.] [1. 4.]] >>> mindspore.ops.rot90(x, k=1, dims=(1, 0)) == mindspore.ops.rot90(x, k=-1, dims=(0,1)) Tensor(shape=[3, 2], dtype=Bool, value= [[ True, True], [ True, True], [ True, True]]) >>> # when input array has ndim>2 >>> x = mindspore.tensor([[[1, 2, 3], ... [4, 5, 6]], ... [[7, 8, 9], ... [10, 11, 12]]], mindspore.float32) >>> output = mindspore.ops.rot90(x, k=1, dims=(2, 1)) >>> print(output) [[[ 4. 1.] [ 5. 2.] [ 6. 3.]] [[10. 7.] [11. 8.] [12. 9.]]]