其他分享
首页 > 其他分享> > mxnet随笔-repeat

mxnet随笔-repeat

作者:互联网

 # -*- coding: utf-8 -*-
"""
Spyder Editor

This is a temporary script file.
"""
import mxnet as mx

x = mx.nd.array([[ 1, 2],[ 3, 4]])

print mx.nd.repeat(x, repeats=2) 

print mx.nd.repeat(x, repeats=2, axis=1) 

print mx.nd.repeat(x, repeats=2, axis=0)
print mx.nd.repeat(x, repeats=2, axis=-1)

[1. 1. 2. 2. 3. 3. 4. 4.]
<NDArray 8 @cpu(0)>

[[1. 1. 2. 2.]
[3. 3. 4. 4.]]
<NDArray 2x4 @cpu(0)>

[[1. 2.]
[1. 2.]
[3. 4.]
[3. 4.]]
<NDArray 4x2 @cpu(0)>

[[1. 1. 2. 2.]
[3. 3. 4. 4.]]
<NDArray 2x4 @cpu(0)>

标签:repeat,mxnet,随笔,nd,print,repeats,mx,axis
来源: https://blog.csdn.net/AI_LX/article/details/89669659