时间戳

首页 > 解决方案 > 在 TensorFlow 中切片张量的困难

python - 在 TensorFlow 中切片张量的困难

问题描述

我正在尝试将输入序列张量减少为序列的最后一个元素，但它似乎没有应用更改：

tf.print(f'EncoderLayer.call.query: {query}')
newquery = query[:][-1:][:]
tf.print(f'EncoderLayer.call.newquery: {newquery}')

结果：

EncoderLayer.call.query: Tensor("test_model_36/dense_291/BiasAdd:0", shape=(None, 512, 512), dtype=float32)
EncoderLayer.call.newquery: Tensor("test_model_36/decoder_layer_0/strided_slice_2:0", shape=(None, 512, 512), dtype=float32)

如果有人好奇，我这样做是为了实现本文中的一个想法：Last Query Transformer RNN for Knowledge tracking

标签： pythontensorflow

解决方案

替换这一行：

newquery = query[:, -1:]

推荐阅读