public final class BroadcastUtils extends Object
Constructor and Description |
---|
BroadcastUtils() |
Modifier and Type | Method and Description |
---|---|
static List<int[]> |
broadcastShapes(int[]... shapes)
Take in shapes and broadcast them to same rank
|
static List<int[]> |
broadcastShapesToMax(int[] maxShape,
int[]... shapes)
Take in shapes and broadcast them to maximum shape
|
static int[][] |
calculateBroadcastShapes(int[] oldShape,
int size,
int... newShape)
Calculate shapes for broadcasting
|
static Dataset[] |
convertAndBroadcast(Object... objects)
Converts and broadcast all objects as datasets of same shape
|
static int[] |
createBroadcastStrides(Dataset a,
int[] broadcastShape)
Create a stride array from a dataset to a broadcast shape
|
static int[] |
createBroadcastStrides(int isize,
int[] oShape,
int[] oStride,
int[] broadcastShape)
Create a stride array from a dataset to a broadcast shape
|
static int[] |
padShape(int[] shape,
int padding)
Pad shape by prefixing with ones
|
public BroadcastUtils()
public static int[][] calculateBroadcastShapes(int[] oldShape, int size, int... newShape)
oldShape
- size
- newShape
- public static int[] padShape(int[] shape, int padding)
shape
- padding
- public static List<int[]> broadcastShapes(int[]... shapes)
shapes
- public static List<int[]> broadcastShapesToMax(int[] maxShape, int[]... shapes)
maxShape
- shapes
- public static int[] createBroadcastStrides(Dataset a, int[] broadcastShape)
a
- datasetbroadcastShape
- public static int[] createBroadcastStrides(int isize, int[] oShape, int[] oStride, int[] broadcastShape)
isize
- oShape
- original shapeoStride
- original stridebroadcastShape
- public static Dataset[] convertAndBroadcast(Object... objects)
objects
- Copyright © 2014–2019 Eclipse Foundation. All rights reserved.