标签： python tensorflow
我有一个张量字典。每个形状为（64，128）或（64，1，128）。称之为tensor_dict
我有一个int32的张量，可以用作tensor_dict的索引（键）。形状为（64，50）或（64，50，1）。称它为tensor_keys
我想创建一个新的张量形状（64、50、128）。它应该使用tensor_keys中的索引作为tensor_dict的键来创建这个新的张量。
我想我可能需要使用assign，但不确定如何做到这一点