tf.slice(inputs, begin, size, name)是tensorflow中对向量inputs的切片，begin表示每个维度要抽取的开始位置，size表示从开始位置往后要抽取的元素个数，若某个维度size为-1则表示抽取对应维度的所有元素。
例子
import tensorflow as tf
import numpy as np
xyz = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9],
[10, 11, 12]])
points = np.array([[11, 22, 33],
[44, 55, 66],
[77, 88, 99],
[1010, 1111, 1212]])
point_cloud = np.concatenate([xyz, points], axis=-1)
point_cloud = np.expand_dims(point_cloud, axis=0)
print(point_cloud.shape)
l0_xyz = tf.slice(point_cloud, [0, 0, 0], [-1, -1, 3])
l0_points = tf.slice(point_cloud, [0, 0, 3], [-1, -1, 3])
输出
print(l0_xyz)
>>> tf.Tensor(
[[[ 1 2 3]
[ 4 5 6]
[ 7 8 9]
[10 11 12]]], shape=(1, 4, 3), dtype=int32)
print(l0_points)
>>> tf.Tensor(
[[[ 11 22 33]
[ 44 55 66]
[ 77 88 99]
[1010 1111 1212]]], shape=(1, 4, 3), dtype=int32)
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
