neureka.fluent.building.states

Interface WithShapeOrScalarOrVectorOnDevice<V>

All Superinterfaces:

WithShapeOrScalarOrVector<V>, WithShapeOrScalarOrVectorTensor<V>

All Known Implementing Classes:

NdaBuilder

public interface WithShapeOrScalarOrVectorOnDevice<V>
extends WithShapeOrScalarOrVectorTensor<V>

Method Summary

Modifier and Type	Method and Description
WithShapeOrScalarOrVectorTensor<V>	on(Device<? super V> device) Use this to specify the type onto which the tensor should be stored.

Methods inherited from interface neureka.fluent.building.states.WithShapeOrScalarOrVectorTensor

scalar, vector, vector, vector, withShape, withShape

Method Detail

on

WithShapeOrScalarOrVectorTensor<V> on(Device<? super V> device)

Use this to specify the type onto which the tensor should be stored.

Parameters:

device - The Device which should host the tensor built by this builder.

Returns:

The next fluent builder API step, which requires the definition of the shape of the tensor.