Can input like is (batch,node,feature)?

[280777510]

Hello, GPU-accelerated version of Kmeans, can pass in a tensor whose shape is 3 dimensional? For example, (3,758,32) is (batch,node,feature), and the batch is expected to be parallel.