码迷,mamicode.com
首页 > Web开发 > 详细

mxnet-broadcast,shape,ndim

时间:2018-11-13 20:38:02      阅读:447      评论:0      收藏:0      [点我收藏+]

标签:script   broadcast   file   utf-8   temp   rip   odi   int   port   

# -*- coding: utf-8 -*- """ Spyder Editor This is a temporary script file. """ import mxnet as mx import numpy as np x = mx.nd.arange(0,3).reshape((1,3,1)) print x y = x.broadcast_to((2,3,3)) print y x = mx.nd.array([1, 2, 3, 4]) print x.ndim x = mx.nd.array([[1, 2], [3, 4]]) print x.ndim print y.shape

shape:形状

ndim:维度

broadcast:广播,但不改变维度

[[[0.]
[1.]
[2.]]]
<NDArray 1x3x1 @cpu(0)>

[[[0. 0. 0.]
[1. 1. 1.]
[2. 2. 2.]]

[[0. 0. 0.]
[1. 1. 1.]
[2. 2. 2.]]]
<NDArray 2x3x3 @cpu(0)>
1
2
(2L, 3L, 3L)

mxnet-broadcast,shape,ndim

标签:script   broadcast   file   utf-8   temp   rip   odi   int   port   

原文地址:http://blog.51cto.com/13959448/2316492

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!