open3d.ml.tf.models.voxel_pooling_grad

open3d.ml.tf.models.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.