open3d.ml.tf.ops.voxel_pooling_grad¶
-
open3d.ml.tf.ops.
voxel_pooling_grad
(positions, features, voxel_size, pooled_positions, pooled_features_gradient, position_fn='average', feature_fn='average', name=None)¶ Gradient for features in VoxelPooling. For internal use only.
- Parameters
positions – A Tensor. Must be one of the following types: float32, float64.
features – A Tensor. Must be one of the following types: float32, float64, int32, int64.
voxel_size – A Tensor. Must have the same type as positions.
pooled_positions – A Tensor. Must have the same type as positions.
pooled_features_gradient – A Tensor. Must have the same type as features.
position_fn – An optional string from: “average”, “nearest_neighbor”, “center”. Defaults to “average”.
feature_fn – An optional string from: “average”, “nearest_neighbor”, “max”. Defaults to “average”.
name – A name for the operation (optional).
- Returns
A Tensor. Has the same type as features.