<!--
{
  "availability" : [
    "iOS: 16.1.0 -",
    "iPadOS: 16.1.0 -",
    "macCatalyst: 16.1.0 -",
    "macOS: 13.0.0 -",
    "tvOS: 16.1.0 -",
    "visionOS: 1.0.0 -"
  ],
  "documentType" : "symbol",
  "framework" : "MetalPerformanceShadersGraph",
  "identifier" : "/documentation/MetalPerformanceShadersGraph/MPSGraph/spaceToBatch(_:spatialAxes:batchAxis:blockDimensions:usePixelShuffleOrder:name:)",
  "metadataVersion" : "0.1.0",
  "role" : "Instance Method",
  "symbol" : {
    "kind" : "Instance Method",
    "modules" : [
      "Metal Performance Shaders Graph",
      "MetalPerformanceShadersGraph"
    ],
    "preciseIdentifier" : "c:objc(cs)MPSGraph(im)spaceToBatchTensor:spatialAxes:batchAxis:blockDimensions:usePixelShuffleOrder:name:"
  },
  "title" : "spaceToBatch(_:spatialAxes:batchAxis:blockDimensions:usePixelShuffleOrder:name:)"
}
-->

# spaceToBatch(_:spatialAxes:batchAxis:blockDimensions:usePixelShuffleOrder:name:)

Creates a space-to-batch operation and returns the result tensor.

```
func spaceToBatch(_ tensor: MPSGraphTensor, spatialAxes: [NSNumber], batchAxis: Int, blockDimensions: [NSNumber], usePixelShuffleOrder: Bool, name: String?) -> MPSGraphTensor
```

## Parameters

`tensor`

The input tensor.

`spatialAxes`

The axes that define the dimensions containing the spatial blocks.

`batchAxis`

The axis that defines the destination dimension, where to copy the blocks.

`blockDimensions`

An array of numbers that defines the size of the rectangular spatial sub-block.

`usePixelShuffleOrder`

A parameter that controls layout of the sub-blocks within the batch dimension.

`name`

The name for the operation.

## Return Value

A valid MPSGraphTensor object.

## Discussion

This operation outputs a copy of the `input` tensor, where values from the
`spatialAxes` (for `usePixelShuffleOrder=YES` 1,2 or 3 axes supported, otherwise
limited only by `MPSNDArray` rank limitations) dimensions are moved in spatial blocks with
rectangular size defined by `blockDimensions` to the `batchAxis` dimension.
Use the `usePixelShuffleOrder` parameter  to control how the data within spatial blocks is ordered
in the `batchAxis` dimension: with `usePixelShuffleOrder=YES` MPSGraph stores
the values of the spatial blocks contiguosly within the `batchAxis` dimension, whereas
otherwise they are stored interleaved with existing values in the `batchAxis` dimension.
Note: This operation is the inverse of
[`batchToSpace(_:spatialAxes:batchAxis:blockDimensions:usePixelShuffleOrder:name:)`](/documentation/MetalPerformanceShadersGraph/MPSGraph/batchToSpace(_:spatialAxes:batchAxis:blockDimensions:usePixelShuffleOrder:name:)).
Note: This operation is a generalization of
[`space(toDepth2DTensor:widthAxis:heightAxis:depthAxis:blockSize:usePixelShuffleOrder:name:)`](/documentation/MetalPerformanceShadersGraph/MPSGraph/space(toDepth2DTensor:widthAxis:heightAxis:depthAxis:blockSize:usePixelShuffleOrder:name:)).

---

Copyright &copy; 2026 Apple Inc. All rights reserved. | [Terms of Use](https://www.apple.com/legal/internet-services/terms/site.html) | [Privacy Policy](https://www.apple.com/privacy/privacy-policy)