如何创建一个带有一个正弦元素而另一个为余弦波的二维张量？

问题描述

我在做

train_data = np.array([np.sin(time),np.cos(time)])

这给了我形状为(2,4000)的东西。我认为我需要将其设置为(4000,2)，因此对于每个时间步，我都可以得到罪过和败笔。

我该怎么做？

解决方法

您可以使用numpy转置来切换数组。

尝试一下：

train_data = np.array([np.sin(time),np.cos(time)]).T

这就是我要做的（4000,2）。

import numpy as np
time = np.arange(4000)
train_data = np.array([np.sin(time),np.cos(time)])
print (train_data.shape)
print (train_data.T.shape)

此输出为：

(2,4000)
(4000,2)