假设我有张量:
tensor = tf.constant(
[[[0.05340263, 0.27248233, 0.49127685, 0.07926575, 0.96054204],
[0.50013988, 0.05903472, 0.43025479, 0.41379231, 0.86508251],
[0.02033722, 0.11996034, 0.57675261, 0.12049974, 0.65760677],
[0.71859089, 0.22825203, 0.64064407, 0.47443116, 0.64108334]],
[[0.18813498, 0.29462021, 0.09433628, 0.97393446, 0.33451445],
[0.01657461, 0.28126666, 0.64016929, 0.48365073, 0.26672697],
[0.9379696 , 0.44648103, 0.39463243, 0.51797975, 0.4173626 ],
[0.89788558, 0.31063058, 0.05492096, 0.86904097, 0.21696292]],
[[0.07279436, 0.94773635, 0.34173115, 0.7228713 , 0.46553334],
[0.61199848, 0.88508141, 0.97019517, 0.61465985, 0.48971128],
[0.53037002, 0.70782324, 0.32158754, 0.2793538 , 0.62661128],
[0.52787814, 0.17085317, 0.83711126, 0.40567032, 0.71386498]]])
形状是(3,4,5)
我想将其切片以返回一个新的形状张量(3,5),带有一个给定的1D张量,其值指示要检索的位置,例如:
index_tensor = tf.constant([2,1,3])
产生一个新的张量,如下所示:
[[0.02033722, 0.11996034, 0.57675261, 0.12049974, 0.65760677],
[0.01657461, 0.28126666, 0.64016929, 0.48365073, 0.26672697],
[0.52787814, 0.17085317, 0.83711126, 0.40567032, 0.71386498]]
也就是说,沿着第二个维度,从索引2、1和3中获取项。 类似于:
tensor[:,x,:]
除了这将只给我在沿维度的索引'x'的项目,我希望它是灵活的。你知道吗
这能做到吗?你知道吗
这会像这样打印值。你知道吗
您可以使用
tf.one_hot()
来屏蔽index_tensor
。你知道吗然后通过
tf.boolean_mask()
得到结果。你知道吗相关问题 更多 >
编程相关推荐