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

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

我在做

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)