如何在python3中初始化特定形状的张量

时间:2020-03-14 03:31:31

标签: python python-3.x numpy

我需要初始化的张量的形状为(3,3,2)。 因此,我需要初始化以下3种形状为(3,2)的矩阵:

第一个矩阵:[[x1,x2],[0,0],[0,0]]

第二个矩阵:[[0,0],[x1,x2],[0,0]]

第三矩阵:[[0,0],[0,0],[x1,x2]]

我得到了x(numpy array)作为输入。 我在弄清楚使用numpy初始化张量的循环时遇到了麻烦。

   y= np.zeros((3, 3, 2))
   for i in range(3):
      for j in range(3):
         for k in range(2):
             y[i,j,k] = ?

我认为给定x,这是错误的做法。任何帮助表示赞赏!

1 个答案:

答案 0 :(得分:0)

rename 's/bre//' bre*