pytorch中想要加入这样一个神经网络,torch.nn

先定义一个神经网络
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进行调控.,请问如何再加一个神经网络,请用代码解析.要求见下.

  1. 要求在 Y 经过新网络后,在第三维数轴上,也是 2 的那个位置.第一个数字不做要求或者保持不变,而第二数字则首先大于0.
  2. 最好第2个数字小于 第一个数字的绝对值.这个尽量就好
  3. 举例:BATCH_SIZE = 3

img

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>]]

是循环值吗？