先定义一个神经网络fc=torch.nn.Linear(s_dim, 2)
在forward
运行中:Y=fc(X)
也就是X是输入数据,Y是经过nn.Linear
后的输出.
假设X.shape=(BATCH_SIZE, 5, s_dim)是一个三维数组.
Y.shape=(BATCH_SIZE, 5, 2)也是一个三维数组.
接下来,我想要再加一个网络对Y进行调控.,请问如何再加一个神经网络,请用代码解析.要求见下.
- 要求在 Y 经过新网络后,在第三维数轴上,也是 2 的那个位置.第一个数字不做要求或者保持不变,而第二数字则首先大于0.
- 最好第2个数字小于 第一个数字的绝对值.这个尽量就好
- 举例:BATCH_SIZE = 3
Y = [[[0,0],[0,0],[8,3],[0,0],<font color=green>[4,-1] </font>],
[[0,0], [0,0], <font color=orange>[-8,-2]</font>, [0,0],[4,1]],
[ [10,0], <font color=purple>[-7,-1], [-8,-2], [-1,2],[4,-1]</font>]]
NEW_Y=[[[0,0], [0,0], [8,3], [0,0],<font color=green>[4, 1] </font> ],
[[0,0], [0,0], <font color=orange>[-8,-2]</font>, [0,0],[4,1]],
[10,0], <font color=purple>[-7, 1], [-8, 2], <font color=red>[-1,1]</font>,[4,1]</font>]]