Method NewTensor
NewTensor(TensorShape, DataType, AllocScope)
Allocates a new tensor with the internal allocator.
Declaration
public override Tensor NewTensor(TensorShape shape, DataType dataType, AllocScope scope)
Parameters
Type | Name | Description |
---|---|---|
TensorShape | shape | The shape of the tensor. |
DataType | dataType | The data type of the tensor. |
AllocScope | scope | Whether the allocated tensor is internal to the layer or used as an output. |
Returns
Type | Description |
---|---|
Tensor | The allocated tensor. |