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.