【发布时间】:2018-05-23 13:15:45
【问题描述】:
假设,我有一个形状为 [a, b, c] 的 3 维张量。 我想在模型运行时提取一维,例如[4, 2, c],所以我最终会得到一个一维张量 [c]。
a 和 b 的参数存储在形状为 [a, 1] 和 [b, 1] 的不同张量中,因此不能使用 tf.slice,因为 tf.slice 只接受一维标量张量。
有什么想法吗?
谢谢!
【问题讨论】:
-
a和b的参数存储在不同的张量中是什么意思?这些参数是做什么用的?你能举一个python/numpy的例子吗?
标签: numpy tensorflow matrix-multiplication